updated sync bn

Lightning-AI · Aug 5, 2020 · 66f948b · 66f948b
1 parent ed391e1
commit 66f948b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pl_examples/basic_examples/sync_bn.py b/pl_examples/basic_examples/sync_bn.py
@@ -145,7 +145,7 @@ def main(args, datamodule, bn_outputs):
         distributed_backend=args.dist_backend,
         max_epochs=args.epochs,
         max_steps=args.steps,
-        sync_bn=args.bn_sync,
+        sync_batchnorm=args.bn_sync,
         num_sanity_val_steps=0,
         replace_sampler_ddp=False,
     )