-
Notifications
You must be signed in to change notification settings - Fork 6
Open
Description
I have commited an implementation of DC-Info-GAN in the repository here. I've tried to write it in such a way to allow easy experimentation with this from now on. Please comment here if there are any questions and suggestions!
Right now I have only integrated the proper loss function for the discrete distributions, and this is working. For continuous variables this is not working great yet. These are the results for training 10 epochs:
Here is my next steps:
- Make Generation of images adapt to different latent specs
- Implement Mutual Information Loss for continous latent variables. I had a discussion with Ben on this and I think I remember that maybe MSE is actually enough here... More on this soon.
- More experimentation with different latent specs / network configurations / loss functions, etc.
- Work towards integrating with BEGAN.
- Integrating TensorBoard for better evaluation.
