Commit Graph

8 Commits

Author SHA1 Message Date
William Falcon 3f76152470 added on_after_backward 2019-07-21 18:23:48 -04:00
William Falcon f6416f737d added grad hook 2019-07-21 18:15:58 -04:00
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
William Falcon 4b0b7e5ea3 if return -1 from a hook that loop stopps 2019-04-21 13:40:32 -04:00
William Falcon e89da15f18 if return -1 from a hook that loop stopps 2019-04-21 13:38:50 -04:00
William Falcon 398b709b76 fixex imports 2019-04-21 13:12:42 -04:00
Shreyas Bapat 18b0c5a122 Add src, docs and other important folders 2019-04-03 22:16:02 +05:30
William Falcon 2117485550 updated lib name 2019-03-30 21:45:16 -04:00