Skip to content

release v0.1.0
Compare
Choose a tag to compare
@benoitmartin88 benoitmartin88 released this 16 Sep 15:18
· 18 commits to master since this release
v0.1.0
9277077

Changelog:

  • ModuleTrainer
  • Early stopping: stop training after a period of stagnation
  • Checkpointing: save model and estimator at regular intervals
  • CSV file writer to output logs
  • Several metrics are available: all default PyTorch loss functions, Accuracy, MAE
  • Progress bar from console
  • SIGINT handling: handle CTRL-C
  • Model's data type (float32, float64)
Assets 2
Loading