b00991efd8
* Added changeable extension variable for model checkpoints * Removed whitespace * Removed the last bit of whitespace * Wrote tests for FILE_EXTENSION * Fixed formatting issues * More formatting issues * Simplify test by just using defaults * Formatting to PEP8 * Added dummy class that inherits ModelCheckpoint; run only one batch instead of epoch for integration test * Fixed too much whitespace formatting * some changes Co-authored-by: rohitgr7 <rohitgr1998@gmail.com> |
||
---|---|---|
.. | ||
__init__.py | ||
test_checkpoint_callback_frequency.py | ||
test_model_checkpoint.py | ||
test_torch_saving.py |