lightning/pytorch_lightning/core
ananthsub 1eb1d17e25
Add trainer attribute to datamodule (#3749)
* 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

* Add attribute to datamodule for trainer
2020-10-01 00:41:19 +05:30
..
__init__.py document lightiningmodule better (#2920) 2020-08-11 19:39:43 -04:00
datamodule.py Add trainer attribute to datamodule (#3749) 2020-10-01 00:41:19 +05:30
decorators.py added copyright notices (#3062) 2020-08-19 22:03:22 -04:00
grads.py added copyright notices (#3062) 2020-08-19 22:03:22 -04:00
hooks.py Support checkpoint hooks on data module (#3563) 2020-09-29 19:51:44 +02:00
lightning.py Drop all result docs. Make the separation between flow and logging clear (#3744) 2020-09-30 08:31:16 -04:00
memory.py Refactor GPUStatsMonitor to improve training speed (#3257) 2020-09-04 06:02:16 -04:00
saving.py Use full yaml loader (load is deprecated) (#3357) 2020-09-07 10:00:35 +02:00
step_result.py tests for val step flow and logging (#3731) 2020-09-29 22:12:56 -04:00