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