Revert D30888794: [Model Averaging] Simplify PostLocalSGD Optimizer API
authorAlban Desmaison <albandes@fb.com>
Thu, 16 Sep 2021 13:36:29 +0000 (06:36 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 16 Sep 2021 13:39:57 +0000 (06:39 -0700)
commit8800a8b4281ca80f35dc338bccb40a678710509d
tree8d52a3db598b6f835185984f1ddf8f05f9e8d6a0
parent83878e19ff2af39443016beb31952476c9f049d0
Revert D30888794: [Model Averaging] Simplify PostLocalSGD Optimizer API

Test Plan: revert-hammer

Differential Revision:
D30888794 (https://github.com/pytorch/pytorch/commit/3d312b3b8ee90f8b289c7d5601a13d0521b46b7e)

Original commit changeset: 21261b480f6b

fbshipit-source-id: 87abb7e8cd9ecaac909ec6c3ee053fa7c4ae1975
torch/distributed/optim/post_localSGD_optimizer.py
torch/testing/_internal/distributed/distributed_test.py