William Falcon 6d1d5ef68e set dp as default backend		2019-07-18 11:45:55 -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	set dp as default backend	2019-07-18 11:45:55 -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