pytorch/torch/distributed/algorithms
PyTorch MergeBot 1d233b8f50 Revert "Make nn.Module state_dict load_state_dict pre-hook and state_dict post hook public (#126704)"
This reverts commit c38b3381a1.

Reverted https://github.com/pytorch/pytorch/pull/126704 on behalf of https://github.com/clee2000 due to broke internal typecheck D58394110 (which probably means the code wouldn't work either but I guess it didn't run on the diff). Probably an easy fix? ([comment](https://github.com/pytorch/pytorch/pull/126704#issuecomment-2161299193))
2024-06-11 17:45:20 +00:00
..
_checkpoint Revert "Make nn.Module state_dict load_state_dict pre-hook and state_dict post hook public (#126704)" 2024-06-11 17:45:20 +00:00
_comm_hooks Flip default value for mypy disallow_untyped_defs [5/11] (#127842) 2024-06-08 18:49:18 +00:00
_optimizer_overlap Flip default value for mypy disallow_untyped_defs [5/11] (#127842) 2024-06-08 18:49:18 +00:00
_quantization Flip default value for mypy disallow_untyped_defs [5/11] (#127842) 2024-06-08 18:49:18 +00:00
ddp_comm_hooks Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-06-10 20:44:42 +00:00
model_averaging Flip default value for mypy disallow_untyped_defs [5/11] (#127842) 2024-06-08 18:49:18 +00:00
__init__.py
join.py Flip default value for mypy disallow_untyped_defs [5/11] (#127842) 2024-06-08 18:49:18 +00:00