Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix KeyedOptimizer init state test (#1874)
Summary: Pull Request resolved: #1874 `KeyedOptimizer.test_init_state` test started failing since upstream pytorch changes: pytorch/pytorch#122349 (and later follow up: pytorch/pytorch#123757), which only initializes the state for param groups if momentum is enabled for SGD. Updating unit test to enable momentum fixes it. Also adding a new unit test to check state if momentum is disabled. Reviewed By: henrylhtsang Differential Revision: D56076424 fbshipit-source-id: e3d5ae063a5187d2d8702ad7f0bb4b2791b954fe
- Loading branch information