ananthsub 3dcf7130c5 Support checkpoint hooks on data module (#3563) ... * Split out changes from #3563 to make that PR easier to review. This formats the file according to the Black formatter * Store a reference to the trainer on the datamodule Fixes #3682 * Update data_connector.py * Update data_connector.py * Update test_datamodules.py * Split out changes from #3563 to make that PR easier to review. This formats the file according to the Black formatter * support checkpoint hooks for datamodule refactor on_{save/load}_checkpoint to a separate hook class that both the lightning module and data module inherit add spots in callback connector to call new datamodule hooks if available * hooks formatting * Update hooks.py * Update checkpoint_connector.py * Update lightning.py * update based on upstream/master checkout upstream/master * Update checkpoint_connector.py * add tests * undo format revert * Updated CHANGELOG.md * add checkpoint hooks * add Dict type * import CheckpointHooks		2020-09-29 19:51:44 +02:00
..
__init__.py	Refactor model summary + generalize example input array (#1773)	2020-06-15 17:05:58 -04:00
test_datamodules.py	Support checkpoint hooks on data module (#3563)	2020-09-29 19:51:44 +02:00
test_decorators.py	cleaning tests (#2201)	2020-06-15 22:03:40 -04:00
test_memory.py	fix dtype conversion of example_input_array in model summary (#2510)	2020-07-05 07:17:22 -04:00
test_results.py	use tmpdir in tests when writing predictions to disk (#3561)	2020-09-23 07:44:15 -04:00