pytorch/torch/distributed/algorithms
2023-11-14 18:52:41 +00:00
..
_checkpoint Fix docstring errors in default_hooks.py, optimizer_overlap.py, checkpoint_wrapper.py, copy.py, benchmark_ddp_rpc.py, utils.py, dependency.py, phony.py, pipeline.py, checkpoint.py, worker.py, batchnorm.py, quantization.py (#113511) 2023-11-14 18:52:41 +00:00
_comm_hooks Fix docstring errors in default_hooks.py, optimizer_overlap.py, checkpoint_wrapper.py, copy.py, benchmark_ddp_rpc.py, utils.py, dependency.py, phony.py, pipeline.py, checkpoint.py, worker.py, batchnorm.py, quantization.py (#113511) 2023-11-14 18:52:41 +00:00
_optimizer_overlap Fix docstring errors in default_hooks.py, optimizer_overlap.py, checkpoint_wrapper.py, copy.py, benchmark_ddp_rpc.py, utils.py, dependency.py, phony.py, pipeline.py, checkpoint.py, worker.py, batchnorm.py, quantization.py (#113511) 2023-11-14 18:52:41 +00:00
_quantization Fix docstring errors in default_hooks.py, optimizer_overlap.py, checkpoint_wrapper.py, copy.py, benchmark_ddp_rpc.py, utils.py, dependency.py, phony.py, pipeline.py, checkpoint.py, worker.py, batchnorm.py, quantization.py (#113511) 2023-11-14 18:52:41 +00:00
ddp_comm_hooks Enable fused optimizer for DP (#98270) 2023-04-13 20:16:32 +00:00
model_averaging Convert logging f-strings to use % format, part four (#98705) 2023-04-11 13:17:59 +00:00
__init__.py
join.py fix docstring issues in torch.distributed (#113337) 2023-11-13 19:37:29 +00:00