lightning/pytorch_lightning/accelerators
Jirka Borovec 4354690e55
add apex test (#2921)
* add apex test

* rename

* level

* events

* wrap

* evt

* miss

* apex

* apex

* apex

* apex

* apex

* apex

* Update tests/models/test_amp.py

Co-authored-by: William Falcon <waf2107@columbia.edu>

* notes

* notes

Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-08-13 10:03:13 -04:00
..
__init__.py clean imports (#2867) 2020-08-08 00:33:51 +02:00
cpu_backend.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
ddp2_backend.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
ddp_backend.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
ddp_spawn_backend.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
dp_backend.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
gpu_backend.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
tpu_backend.py clean imports (#2867) 2020-08-08 00:33:51 +02:00