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

split trainer tests (#956)

2020-02-26 20:31:40 -05:00

test_checks.py

Removed test_dataloader call in check_testing_model_configuration (#1670)

2020-05-12 00:08:07 -04:00

test_dataloaders.py

remove extra kwargs from Trainer init (#1820)

2020-05-17 09:14:54 -04:00

test_lr_finder.py

Bugfix: accumulation and suggestion for learning rate finder (#1801)

2020-05-13 14:40:44 -04:00

test_optimizers.py

Tests: refactor cleanup (#1744)

2020-05-10 13:15:28 -04:00

test_trainer_cli.py

extend arg parser (#1842)

2020-05-14 17:56:11 -04:00

test_trainer_tricks.py

Tests: refactor cleanup (#1744)

2020-05-10 13:15:28 -04:00

test_trainer.py

Fix save_weights_only flag in ModelCheckpoint (#1780)

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