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
..
__init__.py	fixex imports	2019-04-21 13:12:42 -04:00
grads.py	fixex imports	2019-04-21 13:12:42 -04:00
hooks.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
memory.py	fixex imports	2019-04-21 13:12:42 -04:00
model_saving.py	clean up dead code	2019-07-03 17:09:39 -04:00
optimization.py	fixex imports	2019-04-21 13:12:42 -04:00
root_module.py	fixed metrics request not forced anymore	2019-07-15 13:03:08 -04:00