Yi Wang 1b4bb3691c [Gradient Compression] Update _powerSGD_comm_hook_wrapper to only expose 2 most critical hyperparameters (#55295) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55295 Update `_powerSGD_comm_hook_wrapper` to only expose 2 most critical hyperparameters, to make this API more clear to any future user (although the second hyperparameter `start_powerSGD_iter` is not in use yet). Test Plan: waitforbuildbot Reviewed By: shuyingsunshine21 Differential Revision: D27561734 fbshipit-source-id: b661981cc033b109f4f2fc92b435567a184a7fb5		2021-04-06 01:29:10 -07:00
..
ddp_comm_hooks	[Gradient Compression] Update _powerSGD_comm_hook_wrapper to only expose 2 most critical hyperparameters (#55295)	2021-04-06 01:29:10 -07:00
__init__.py	[Gradient Compression] Add unit tests that test default Python comm hook implementations (#47158)	2020-11-06 00:28:09 -08:00