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	More robust way of collecting init argument names for LightningModules (#3066)	2020-08-20 07:19:11 -04:00
dataloaders.py	Bugfix/_has_len (#2307)	2020-06-26 09:31:08 -04:00
datamodules.py	Support checkpoint hooks on data module (#3563)	2020-09-29 19:51:44 +02:00
datasets.py	clean tests imports (#2834)	2020-08-06 16:58:51 +02:00
deterministic_model.py	[WIP] ref: deprecated results obj, added support for simpler comms (1/n) (#3681)	2020-09-27 23:19:46 -04:00
develop_pipelines.py	Add a more direct test of multi-gpu training working (#2084)	2020-09-29 15:38:09 +02:00
develop_utils.py	fix dp issues + update examples and test examples (#3618)	2020-09-23 00:19:46 -04:00
mixins.py	Fix lr key name in case of param groups (#1719)	2020-05-10 17:05:34 -04:00
model_optimizers.py	Fix batch_outputs with optimizer frequencies (#3229)	2020-09-10 23:01:20 +02:00
model_template.py	More robust way of collecting init argument names for LightningModules (#3066)	2020-08-20 07:19:11 -04:00
model_test_dataloaders.py	Add missing test for "multiple dataloader + percent_check fix" (#2226)	2020-06-23 11:21:24 -04:00
model_test_epoch_ends.py	Add missing test for "multiple dataloader + percent_check fix" (#2226)	2020-06-23 11:21:24 -04:00
model_test_steps.py	ref: result 1/n (make monitor default to checkpoint_on to simplify re… (#3571)	2020-09-20 22:58:43 -04:00
model_train_dataloaders.py	Bugfix/_has_len (#2293)	2020-06-19 23:38:15 -04:00
model_train_steps.py	ref: (results 1/n) enable tracking original metric when step and epoch are both true (#3685)	2020-09-27 22:08:31 -04:00
model_utilities.py	repair CI for Win (#2358)	2020-06-26 21:38:25 -04:00
model_valid_dataloaders.py	Add missing test for "multiple dataloader + percent_check fix" (#2226)	2020-06-23 11:21:24 -04:00
model_valid_epoch_ends.py	remove flake 8 (#3687)	2020-09-27 20:40:02 -04:00
model_valid_steps.py	enable None model checkpoint default (#3669)	2020-09-26 23:14:04 -04:00
models.py	to_torchscript method for LightningModule (#3258)	2020-09-03 20:24:44 +02:00
test_datasets.py	fixing TPU tests (#2632)	2020-07-27 19:07:09 -04:00