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

..

callbacks

Fix save_weights_only flag in ModelCheckpoint (#1780)

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

core

remove obsolete self._device in Trainer (#1849)

2020-05-17 08:20:51 -04:00

loggers

dummy logger (#1836)

2020-05-14 10:34:11 -04:00

logging

Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786)

2020-05-12 09:15:25 +00:00

overrides

Add ddp_cpu backend for testing ddp without GPUs (#1158)

2020-04-15 23:17:31 -04:00

profiler

Profiler summary (#1259)

2020-03-31 08:57:48 -04:00

pt_overrides

Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786)

2020-05-12 09:15:25 +00:00

root_module

Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786)

2020-05-12 09:15:25 +00:00

trainer

Fix save_weights_only flag in ModelCheckpoint (#1780)

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

utilities

Feature: auto scale batch size (#1638)

2020-05-09 08:28:36 -04:00

__init__.py

continue devel (#1793)

2020-05-17 08:30:45 -04:00