Fabio Natanael KeplerandJirka Borovec 8c4c7b105e Fix save_weights_only flag in ModelCheckpoint (#1780) ...

* Add flag to `dump_checkpoint` for only including weights

`ModelCheckpoint` then passes `self.save_weights_only` to the save function.

* Fix tests and add changelog entry

* Add check and descriptive message when training state is restored from a weights only checkpoint

Also add a test for making sure `ModelCheckpoint.save_weights_only` works as expected.

* Fix weights-only test to properly match expected exception

* Apply suggestions from code review

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

2020-05-17 09:24:17 -04:00

..

__init__.py

Learning rate log callback (#1498)

2020-04-30 08:06:41 -04:00

base.py

Progress bar callback (#1450)

2020-04-23 20:46:18 -04:00

early_stopping.py

improve pickle tests for callbacks (#1717)

2020-05-05 14:08:54 -04:00

gradient_accumulation_scheduler.py

changelog (#1616)

2020-04-26 16:11:22 -04:00

lr_logger.py

Fix lr key name in case of param groups (#1719)

2020-05-10 17:05:34 -04:00

model_checkpoint.py

Fix save_weights_only flag in ModelCheckpoint (#1780)

2020-05-17 09:24:17 -04:00

progress.py

enable fast_dev_run without a validation loop (#1779)

2020-05-11 11:30:22 -04:00