William Falcon a41abad5b2 Update trainer.py

2019-07-16 17:02:21 -04:00

..

callbacks

clean up dead code

2019-07-03 17:05:20 -04:00

models

Update trainer.py

2019-07-16 17:02: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

prog bar option

2019-06-27 11:22:13 -04:00