├── .circleci └── config.yml ├── .gitignore ├── CHANGELOG.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── examples └── mnist.py ├── pytorchtrainer ├── __init__.py ├── callback │ ├── __init__.py │ ├── callback.py │ ├── checkpoint.py │ ├── file_writer.py │ ├── metric.py │ └── validation.py ├── main.py ├── metric │ ├── __init__.py │ ├── accuracy.py │ ├── mean_absolute_error.py │ └── torch_loss.py ├── stop_condition │ ├── __init__.py │ └── early_stopping.py ├── trainer.py └── utils.py ├── requirements.txt ├── setup.py └── test ├── __init__.py ├── common.py ├── test_callback.py ├── test_metric.py ├── test_stop_condition.py └── test_trainer.py /.circleci/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/.circleci/config.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/README.md -------------------------------------------------------------------------------- /examples/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/examples/mnist.py -------------------------------------------------------------------------------- /pytorchtrainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/__init__.py -------------------------------------------------------------------------------- /pytorchtrainer/callback/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/callback/__init__.py -------------------------------------------------------------------------------- /pytorchtrainer/callback/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/callback/callback.py -------------------------------------------------------------------------------- /pytorchtrainer/callback/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/callback/checkpoint.py -------------------------------------------------------------------------------- /pytorchtrainer/callback/file_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/callback/file_writer.py -------------------------------------------------------------------------------- /pytorchtrainer/callback/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/callback/metric.py -------------------------------------------------------------------------------- /pytorchtrainer/callback/validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/callback/validation.py -------------------------------------------------------------------------------- /pytorchtrainer/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/main.py -------------------------------------------------------------------------------- /pytorchtrainer/metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/metric/__init__.py -------------------------------------------------------------------------------- /pytorchtrainer/metric/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/metric/accuracy.py -------------------------------------------------------------------------------- /pytorchtrainer/metric/mean_absolute_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/metric/mean_absolute_error.py -------------------------------------------------------------------------------- /pytorchtrainer/metric/torch_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/metric/torch_loss.py -------------------------------------------------------------------------------- /pytorchtrainer/stop_condition/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/stop_condition/__init__.py -------------------------------------------------------------------------------- /pytorchtrainer/stop_condition/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/stop_condition/early_stopping.py -------------------------------------------------------------------------------- /pytorchtrainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/trainer.py -------------------------------------------------------------------------------- /pytorchtrainer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/pytorchtrainer/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | 2 | torch>=1.0.0 3 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/setup.py -------------------------------------------------------------------------------- /test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/test/common.py -------------------------------------------------------------------------------- /test/test_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/test/test_callback.py -------------------------------------------------------------------------------- /test/test_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/test/test_metric.py -------------------------------------------------------------------------------- /test/test_stop_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/test/test_stop_condition.py -------------------------------------------------------------------------------- /test/test_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benoitmartin88/pytorchtrainer/HEAD/test/test_trainer.py --------------------------------------------------------------------------------