Pytorchtrainer

Latest version: v0.2.0

Safety actively analyzes 621673 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

0.2.0

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`

0.1.0

New
- `ModuleTrainer` object
- `EarlyStopping`: stop training after a configurable 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)
- Full use of Pytorch's Cuda support

Links

Releases

Has known vulnerabilities

© 2024 Safety CLI Cybersecurity Inc. All Rights Reserved.