Releases: FrancescoSaverioZuppichini/PyTorch-Deep-Learning-Template
Releases · FrancescoSaverioZuppichini/PyTorch-Deep-Learning-Template
Pytorch Deep Learning Template
First beta release! For now, we use the following tools
- data-augmentation: we included imgaug
- ready to go: by using poutyne a Keras-like framework you don't have to write any train loop.
- torchsummary to show a summary of your models
- reduce the learning rate on a plateau
- auto-saving the best model
- experiment tracking with comet
- logging using python logging module
- a playground notebook to quick test/play around