lightning/tests/models
Adrian Wälchli 3c549e8ae3
Call on_before_zero_grad model hook (#1493)
* call on_before_zero_grad

* update changelog

* add note about overriding both hooks

* added test

* move test_hooks.py to models folder
2020-04-16 12:01:41 -04:00
..
__init__.py CI: split tests-examples (#990) 2020-03-25 07:46:27 -04:00
test_amp.py Fix weights path (#1445) 2020-04-10 12:02:59 -04:00
test_cpu.py Add ddp_cpu backend for testing ddp without GPUs (#1158) 2020-04-15 23:17:31 -04:00
test_gpu.py Fix weights path (#1445) 2020-04-10 12:02:59 -04:00
test_hooks.py Call on_before_zero_grad model hook (#1493) 2020-04-16 12:01:41 -04:00
test_restore.py Fix weights path (#1445) 2020-04-10 12:02:59 -04:00