Skip to content

Conversation

@pierrebernard7
Copy link

Hello there,

First of all thanks for your amazing work on gans.
you made them really easy to understand for newbies.

I added a few things to gan.py and cgan.py (which are the only ones I've used so far) :

  • allow to select which gpu to use with a command line argument "--gpu" 0, 1 or 2 etc.
  • tensorboard visualisation of G and D losses
  • improved version of the MNIST dataset class which I found here : https://gist.github.com/y0ast/f69966e308e549f013a92dc66debeeb 4.
    It bypasses the image resizing which converts the image to PIL object and is slow..
    It's only enabled when setting im_size 28 then (original MNIST size).
    I've seen great improvements (x2 speed)

If you want I can also make those changes to the other gans

Cheers.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant