3c549e8ae3
* call on_before_zero_grad * update changelog * add note about overriding both hooks * added test * move test_hooks.py to models folder |
||
---|---|---|
.. | ||
__init__.py | ||
test_amp.py | ||
test_cpu.py | ||
test_gpu.py | ||
test_hooks.py | ||
test_restore.py |