Nathan Raw bab89b8d21 Add transfer_batch_to_device hook to DataModule (#3038) ... * ✨ add dm to_device logic in trainer * 🔥 remove unnecessary comment * ✨ add to_device logic to datamodule * ✅ add test * updated docs Co-authored-by: William Falcon <waf2107@columbia.edu>		2020-08-20 08:47:11 -04:00
..
__init__.py	document lightiningmodule better (#2920)	2020-08-11 19:39:43 -04:00
datamodule.py	Add transfer_batch_to_device hook to DataModule (#3038)	2020-08-20 08:47:11 -04:00
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	added copyright notices (#3062)	2020-08-19 22:03:22 -04:00
lightning.py	added copyright notices (#3062)	2020-08-19 22:03:22 -04:00
memory.py	flake8 fixes (#3064)	2020-08-20 07:45:22 -04:00
saving.py	Make sure all kwargs come after args in _load_model_state() (#3063)	2020-08-20 07:29:44 -04:00
step_result.py	flake8 fixes (#3064)	2020-08-20 07:45:22 -04:00