A simple boilerplate code (that I use) for training neural networks in Pytorch (with Tensorboard).
examples/ contains
- CIFAR_10 image classification (based on 60 Minute Blitz)
- Supports Tensorboard support
- FashionMNIST classification
TODO:
- Generating new FashionMNIST images
- Semantic segmentation
- UNet