search

martinarjovsky / WassersteinGAN

BSD 3-Clause "New" or "Revised" License
3.2k stars 725 forks source link

Add Tensorboard logging for scalars and images, add FastImageFolder for large folders #55

Open snakers4 opened 6 years ago

snakers4 commented 6 years ago

Thank you for your amazing work and code.

Please consider reviewing / adding the following small ideas to the code:

  1. Tensorboard logging. Tb is an amazing tool that really simplifies the NN training process. Added some reference code for Tb logging + added image visualization during training + scalar visualization.

  2. A simple FastImageFolder class to load large datasets from folders. The pre-built PyTorch class handles very large folders (500k+ images) not really fast.

Thank you!