Cinjon Resnick fbd3873a0f 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

..

sample_model_template

added lightning model docs

2019-06-27 10:05:47 -04:00

__init__.py

added trainer docs

2019-06-27 11:03:53 -04:00

trainer.py

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