Skip to content

Releases: FrancescoSaverioZuppichini/PyTorch-Deep-Learning-Template

Pytorch Deep Learning Template

31 Jan 09:10
Compare
Choose a tag to compare

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