From 66f948b6395da066f262975542daafbf06bb74f6 Mon Sep 17 00:00:00 2001 From: ananyahjha93 Date: Wed, 5 Aug 2020 15:11:27 -0400 Subject: [PATCH] updated sync bn --- pl_examples/basic_examples/sync_bn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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, )