Skip to content

Releases: MathGaron/pytorch_toolbox

prototype

26 Oct 20:57

Choose a tag to compare

prototype Pre-release
Pre-release

Basic functions are implemented:

  • Train loop that supports train and validation dataset
  • Flexible callback system to retrieve data during training
  • Simple checkpoint system (could be more flexible..)
  • Configuration loading (yaml)
  • helpers to write information to console/file/visdom