From b27462f7477c3de86ea8b67a3f2cb6ed46ec6498 Mon Sep 17 00:00:00 2001 From: Jason Senthil Date: Thu, 15 Feb 2024 19:46:01 -0800 Subject: [PATCH] disable sync batchnorm by default (#673) Summary: Pull Request resolved: https://github.com/pytorch/tnt/pull/673 Reviewed By: galrotem Differential Revision: D52677039 fbshipit-source-id: 45ebff243e3b0ce672710d9d298110f39f88d44e --- torchtnt/utils/prepare_module.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchtnt/utils/prepare_module.py b/torchtnt/utils/prepare_module.py index 58caffedf4..740eca961c 100644 --- a/torchtnt/utils/prepare_module.py +++ b/torchtnt/utils/prepare_module.py @@ -77,7 +77,7 @@ class DDPStrategy(Strategy): ] = None # SyncBatchNorm params - sync_batchnorm: bool = True + sync_batchnorm: bool = False @dataclass