pytorch
80f40b17 - [Model Averaging] Periodic model averager (#60320)

Commit
3 years ago
[Model Averaging] Periodic model averager (#60320) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60320 This averager can be used for post-local SGD. ghstack-source-id: 131908011 Test Plan: buck test mode/dev-nosan caffe2/test/distributed:distributed_nccl_fork -- test_periodic_model_averager Reviewed By: rohan-varma Differential Revision: D29249850 fbshipit-source-id: 09675d6bb1edfb8ffbeb94510d91962532d8ca3e
Author
Yi Wang
Parents
Loading