Releases: benoitmartin88/pytorchtrainer
Releases · benoitmartin88/pytorchtrainer
Release v0.2.1
Hotfix
- fix ignored sigint
release 0.2.0
New
- Add
ModuleTrainer.evaluate
method - Add CsvWriter to evaluate method
- Add
filename_transform_function
argument toSaveBestCheckpointCallback
- Metric step method now return the intermediate computed values
- Rename
CsvWriter
'sextra
argument toextra_data_function
CsvWriter
can now be called with theextra_data
extra argument to define the extra data that will be logged- Add
ModuleTrainer.load
method
Change
- Rename
dateset_loader
todataloader
release v0.1.0
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
)