Skip to content

Releases: benoitmartin88/pytorchtrainer

Release v0.2.1

10 Oct 15:13
7d73acd
Compare
Choose a tag to compare

Hotfix

  • fix ignored sigint

release 0.2.0

20 Sep 08:50
2a4d0a3
Compare
Choose a tag to compare

New

  • Add ModuleTrainer.evaluate method
  • Add CsvWriter to evaluate method
  • Add filename_transform_function argument to SaveBestCheckpointCallback
  • Metric step method now return the intermediate computed values
  • Rename CsvWriter's extra argument to extra_data_function
  • CsvWriter can now be called with the extra_data extra argument to define the extra data that will be logged
  • Add ModuleTrainer.load method

Change

  • Rename dateset_loader to dataloader

release v0.1.0

16 Sep 15:18
Compare
Choose a tag to compare

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)