lightning/pytorch_lightning/accelerators
Jirka Borovec a6e7aa7796
allow using apex with any PT version (#2865)
* wip

* setup

* type

* name

* wip

* docs

* imports

* fix if

* fix if

* use_amp

* Apply suggestions from code review

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* Apply suggestions from code review

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* fix tests

* Apply suggestions from code review

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* fix tests

* todos

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2020-08-08 11:07:32 +02:00
..
__init__.py clean imports (#2867) 2020-08-08 00:33:51 +02:00
cpu_backend.py allow using apex with any PT version (#2865) 2020-08-08 11:07:32 +02:00
ddp2_backend.py allow using apex with any PT version (#2865) 2020-08-08 11:07:32 +02:00
ddp_backend.py allow using apex with any PT version (#2865) 2020-08-08 11:07:32 +02:00
ddp_spawn_backend.py allow using apex with any PT version (#2865) 2020-08-08 11:07:32 +02:00
dp_backend.py allow using apex with any PT version (#2865) 2020-08-08 11:07:32 +02:00
gpu_backend.py allow using apex with any PT version (#2865) 2020-08-08 11:07:32 +02:00
tpu_backend.py clean imports (#2867) 2020-08-08 00:33:51 +02:00