lightning/pytorch_lightning/accelerators
William Falcon 18160b81b5
refactored horovod backend (#3122)
2020-08-24 11:13:49 -04:00
..
__init__.py ref: refactored horovod backend (#3121) 2020-08-24 10:35:32 -04:00
base_backend.py Refactor 1: moved tpu xxx_step to backend (#3118) 2020-08-24 07:02:06 -04:00
cpu_backend.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
ddp2_backend.py refactored ddp backend forward (#3119) 2020-08-24 07:33:14 -04:00
ddp_backend.py refactored ddp backend forward (#3119) 2020-08-24 07:33:14 -04:00
ddp_spawn_backend.py refactored ddp backend forward (#3119) 2020-08-24 07:33:14 -04:00
dp_backend.py refactored ddp backend forward (#3119) 2020-08-24 07:33:14 -04:00
gpu_backend.py ref: refactored gpu backend __step (#3120) 2020-08-24 09:22:05 -04:00
horovod_backend.py refactored horovod backend (#3122) 2020-08-24 11:13:49 -04:00
tpu_backend.py refactored ddp backend forward (#3119) 2020-08-24 07:33:14 -04:00