Rohan Varma 8fa5cde3a9 Fix hooks (#71970) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71970 - Provide default arg for power SGD convenience wrapper that matches the main API default Test Plan: CI Reviewed By: H-Huang Differential Revision: D33837457 fbshipit-source-id: 8f4efab4992b3fff09456a18db2c83e087c25bdf (cherry picked from commit 83f52fb3c7)		2022-01-28 23:07:33 +00:00
..
_checkpoint	[FSDP/Checkpoint] Activation offload support in checkpoint_wrapper (#70165)	2021-12-21 10:08:18 -08:00
_optimizer_overlap	[Opt Overlap] Clean up code in _OptimizerHookState (#71620)	2022-01-26 19:33:49 +00:00
ddp_comm_hooks	Fix hooks (#71970)	2022-01-28 23:07:33 +00:00
model_averaging	[LocalSGD] Move feature to Beta, clean up some docs (#71621)	2022-01-21 21:10:42 +00:00
quantization	[BE] minor improvement to dist quantization (#67401)	2021-11-21 23:31:59 -08:00
__init__.py	Make _Join, _Joinable, _JoinHook public (#62605)	2021-08-03 12:20:11 -07:00
join.py	Make _Join, _Joinable, _JoinHook public (#62605)	2021-08-03 12:20:11 -07:00