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 |