Skip to content

Commit

Permalink
updated sync bn
Browse files Browse the repository at this point in the history
  • Loading branch information
@ananyahjha93 @williamFalcon
ananyahjha93 authored and williamFalcon committed Aug 5, 2020
1 parent 07aa2ec commit ed391e1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytorch_lightning/trainer/__init__.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -856,7 +856,7 @@ def on_train_end(self, trainer, pl_module):
trainer = Trainer(row_log_interval=50)
sync_batchnorm
^^^^^^^^^^^^^^^^^
^^^^^^^^^^^^^^
Enable synchronization between batchnorm layers across all GPUs.
Expand Down

0 comments on commit ed391e1

Please sign in to comment.