William Falcon a83588b14e Update trainer.py		2019-07-16 13:12:56 -04:00
..
callbacks	clean up dead code	2019-07-03 17:05:20 -04:00
models	Update trainer.py	2019-07-16 13:12:56 -04:00
pt_overrides	clean up dead code	2019-07-03 16:46:14 -04:00
root_module	add a hook for on_tng_metrics so that users get access to the grad_norm and mem_map dicts.	2019-07-16 12:51:48 -04:00
utils	fixed nccl init	2019-07-12 14:28:49 -04:00
__init__.py	easy import for lightningModule	2019-07-03 18:43:13 -04:00
trainer_main.py	prog bar option	2019-06-27 11:22:13 -04:00