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	Refactor model summary + generalize example input array (#1773)	2020-06-15 17:05:58 -04:00
test_datamodules.py	Add transfer_batch_to_device hook to DataModule (#3038)	2020-08-20 08:47:11 -04: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	Retrieve last logged val from result by key (#3049)	2020-08-19 18:59:14 -04:00