OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
d6b51e4adf
pytorch/torch/nn/parallel
History
Shen Li 8d6a8d2b3f Fix DDP bug in single process multiple device use cases (#36503) 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/36503

Test Plan: Imported from OSS

Differential Revision: D21179274

Pulled By: mrshenli

fbshipit-source-id: 0afce30ae0ddda753d1e240584a0f80df9aec4c2
2020-04-22 15:06:28 -07:00
..
__init__.py Remove DistributedDataParallelCPU as DDP now supports CPU models (#22864) 2019-07-18 12:50:45 -07:00
__init__.pyi fixed replicate typo in torch/nn/parallel/__init__.pyi (#29005) 2019-11-01 08:00:41 -07:00
_functions.py
common_types.pyi Stubs for torch.nn (#19089) 2019-07-01 09:50:17 -07:00
data_parallel.py Update docs about DP and DDP for CUDA (#35063) 2020-03-20 20:06:37 -07:00
data_parallel.pyi Stubs for torch.nn (#19089) 2019-07-01 09:50:17 -07:00
distributed.py Fix DDP bug in single process multiple device use cases (#36503) 2020-04-22 15:06:28 -07:00
distributed.pyi Stubs for torch.nn (#19089) 2019-07-01 09:50:17 -07:00
parallel_apply.py Catch and print exception traceback in parallel_apply() workers (#18055) 2019-07-26 11:41:22 -07:00
parallel_apply.pyi Stubs for torch.nn (#19089) 2019-07-01 09:50:17 -07:00
replicate.py Fix DDP bug in single process multiple device use cases (#36503) 2020-04-22 15:06:28 -07:00
replicate.pyi Stubs for torch.nn (#19089) 2019-07-01 09:50:17 -07:00
scatter_gather.py Use a simpler way to delete recursive function 2019-05-26 12:17:25 -07:00
scatter_gather.pyi Fix typos (#30606) 2019-12-02 20:17:42 -08:00