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