William Falcon 162b9f4f27 set dp as default backend		2019-07-18 11:15:21 -04:00
..
callbacks	clean up dead code	2019-07-03 17:05:20 -04:00
models	set dp as default backend	2019-07-18 11:15:21 -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	…