Jeremy Jordan
|
a5d1176cf6
|
callback method for on_save_checkpoint (#2501)
* initial draft
* fix test
* Update pytorch_lightning/trainer/callback_hook.py
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
* fix tests
* remove old code
* untested upgrade script
* document limitations
* clean up and add tests
* Update pytorch_lightning/trainer/training_io.py
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
* reflect PR comments
* fix formatting
* Update docs/source/callbacks.rst
* clarify docs
* revert change for loading checkpoints
* small edits
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
|
2020-08-28 16:50:52 +02:00 |