pytorch/torch/distributed/algorithms
PyTorch MergeBot fe361dede4 Revert "[FSDP Optim State] Remove checkpoint prefix (#80480)"
This reverts commit 04c50fec1c.

Reverted https://github.com/pytorch/pytorch/pull/80480 on behalf of https://github.com/suo due to Broke master 04c50fec1c, the test failures were not unrelated
2022-07-06 02:43:27 +00:00
..
_checkpoint Revert "[FSDP Optim State] Remove checkpoint prefix (#80480)" 2022-07-06 02:43:27 +00:00
_comm_hooks FSDP communication hook interface for NO_SHARD strategy (#79833) 2022-06-28 08:03:11 +00:00
_optimizer_overlap make fsdp folder to be public (#72084) 2022-02-02 15:50:14 +00:00
_quantization Enable test: distributed/algorithms/quantization/test_quantization (#80097) 2022-07-01 01:32:33 +00:00
ddp_comm_hooks Revert "Added serialization to postlocal_SGD. (#80435)" 2022-06-30 01:34:10 +00:00
model_averaging Add __all__ to various submodules in torch.fx, distributions, distributed, package (#80367) 2022-06-27 21:27:30 +00:00
__init__.py Make _Join, _Joinable, _JoinHook public (#62605) 2021-08-03 12:20:11 -07:00
join.py Add __all__ to various submodules in torch.fx, distributions, distributed, package (#80367) 2022-06-27 21:27:30 +00:00