pytorch
6ee5e490 - [BE][SyncBN] Avoid sync stats in eval mode (#56982)

Commit
3 years ago
[BE][SyncBN] Avoid sync stats in eval mode (#56982) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56982 SyncBatchNorm should behave as a regular BN layer in eval model, this change ensures that this is the case. In particular, the bug was when `track_running_stats=False`, `bn_training` would be set to True in eval mode, but this would trigger a collective sync in syncBN. However, in eval mode syncBN should behave like a regular BN layer and not do this sync. Closes https://github.com/pytorch/pytorch/issues/48988 Ensured with unittest that when used for inference on a single rank, stats sync is not triggered. ghstack-source-id: 127544421 Test Plan: CI Reviewed By: SciPioneer Differential Revision: D27579297 fbshipit-source-id: 26406e2793f0be14f2daa46ae66f97a8494182ed
Author
Parents
Loading