Rohan Varma 04c50fec1c [FSDP Optim State] Remove checkpoint prefix (#80480) ... Remove `_checkpoint_wrapped_module` prefixes when creating keys for optimizer state_dict. Having these does not actually create an issue for optim_state_dict save / load, but we'd like to strip these keys out for downstream code that consumes these APIs typically expecting checkpointing prefixes to not exist (as checkpointing should be a transparent operation which should not change module / parameter names). Pull Request resolved: https://github.com/pytorch/pytorch/pull/80480 Approved by: https://github.com/awgu, https://github.com/fegin		2022-07-06 01:17:58 +00:00
..
_checkpoint	[FSDP Optim State] Remove checkpoint prefix (#80480)	2022-07-06 01:17:58 +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