Releases: MathGaron/pytorch_toolbox
Releases · MathGaron/pytorch_toolbox
prototype
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