diff --git a/pl_examples/basic_examples/sync_bn.py b/pl_examples/basic_examples/sync_bn.py index bb602a74ea89c..35f2cd4088a44 100644 --- 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, )