Sean Naren 98eb736496 Added getstate/setstate method for torch.save serialization (#4127) ... * Added getstate/setstate method for torch.save serialization, added additional Optional Typing to results object * Added tests to ensure torch.save does not fail * Added flags to ensure compatible ddp cpu environment * Removed torch version check due to minimum already being 1.3, reduced epochs for speed * Moved tests to separate file * Update to accelerator, move to ddp_spawn to prevent hanging ddp		2020-10-13 16:47:23 -04:00
..
__init__.py	notices (#4118)	2020-10-13 07:18:07 -04:00
accelerator.py	Added getstate/setstate method for torch.save serialization (#4127)	2020-10-13 16:47:23 -04:00
accelerator_connector.py	notices (#4118)	2020-10-13 07:18:07 -04:00
cpu_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
ddp2_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
ddp_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
ddp_cpu_slurm_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
ddp_cpu_spawn_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
ddp_cpu_torchelastic_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
ddp_slurm_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
ddp_spawn_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
ddp_torchelastic_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
dp_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
gpu_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
horovod_accelerator.py	ref: accelerator names (#4066)	2020-10-11 01:05:14 -04:00
tpu_accelerator.py	clean docs, enable grad clip in manual mode (#4078)	2020-10-11 13:12:35 -04:00