Please consider reviewing / adding the following small ideas to the code:
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.
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 for your amazing work and code.
Please consider reviewing / adding the following small ideas to the code:
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.
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!