lightning/pytorch_lightning/accelerators
Phil e3528afae3
Move optimizer creation after device placement for ddp backends. (#2904)
2020-08-12 06:34:59 -04: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 Move optimizer creation after device placement for ddp backends. (#2904) 2020-08-12 06:34:59 -04:00
ddp_backend.py Move optimizer creation after device placement for ddp backends. (#2904) 2020-08-12 06:34:59 -04:00
ddp_spawn_backend.py Move optimizer creation after device placement for ddp backends. (#2904) 2020-08-12 06:34:59 -04: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