From 951922c735c1e98b596fd9845de25a62fcdc7c73 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 17 Jul 2021 13:07:19 +0200 Subject: [PATCH] Add `--sync-bn` known issue (#4032) * Add `--sync-bn` known issue * Update train.py --- train.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index 205c73d85e20..15c2c356f60e 100644 --- a/train.py +++ b/train.py @@ -217,6 +217,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary # SyncBatchNorm if opt.sync_bn and cuda and RANK != -1: + raise Exception('can not train with --sync-bn, known issue https://github.com/ultralytics/yolov5/issues/3998') model = torch.nn.SyncBatchNorm.convert_sync_batchnorm(model).to(device) logger.info('Using SyncBatchNorm()') @@ -232,9 +233,9 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary # Process 0 if RANK in [-1, 0]: valloader = create_dataloader(val_path, imgsz_val, batch_size // WORLD_SIZE * 2, gs, single_cls, - hyp=hyp, cache=opt.cache_images and not noval, rect=True, rank=-1, - workers=workers, - pad=0.5, prefix=colorstr('val: '))[0] + hyp=hyp, cache=opt.cache_images and not noval, rect=True, rank=-1, + workers=workers, + pad=0.5, prefix=colorstr('val: '))[0] if not resume: labels = np.concatenate(dataset.labels, 0)