lightning/pytorch_lightning/accelerators
Ananya Harsh Jha 9445c800b0
set device to root gpu (#3042)
2020-08-18 19:28:35 -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 ddp fix for trainer.test() + add basic ddp tests (#2997) 2020-08-16 11:19:57 -04:00
ddp_spawn_backend.py ddp fix for trainer.test() + add basic ddp tests (#2997) 2020-08-16 11:19:57 -04:00
dp_backend.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
gpu_backend.py set device to root gpu (#3042) 2020-08-18 19:28:35 -04:00
tpu_backend.py clean imports (#2867) 2020-08-08 00:33:51 +02:00