Skip to content

Commit

Permalink
disable sync batchnorm by default (#673)
Browse files Browse the repository at this point in the history
Summary: Pull Request resolved: #673

Reviewed By: galrotem

Differential Revision: D52677039
  • Loading branch information
JKSenthil authored and facebook-github-bot committed Feb 15, 2024
1 parent 4a57bda commit 3fe1c09
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchtnt/utils/prepare_module.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ class DDPStrategy(Strategy):
] = None

# SyncBatchNorm params
sync_batchnorm: bool = True
sync_batchnorm: bool = False


@dataclass
Expand Down

0 comments on commit 3fe1c09

Please sign in to comment.