Skip to content

Commit a29fba3

Browse files
committed
disable dist_bn when sync_bn active
1 parent 07d0c4a commit a29fba3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

train.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -438,6 +438,7 @@ def main():
438438

439439
# setup synchronized BatchNorm for distributed training
440440
if args.distributed and args.sync_bn:
441+
args.dist_bn = '' # disable dist_bn when sync BN active
441442
assert not args.split_bn
442443
if has_apex and use_amp == 'apex':
443444
# Apex SyncBN used with Apex AMP

0 commit comments

Comments
 (0)