diff --git a/train.py b/train.py index f12ba49f..71438d1b 100644 --- a/train.py +++ b/train.py @@ -76,7 +76,7 @@ def train( if multi_scale: img_size = round((img_size / 32) * 1.5) * 32 # initiate with maximum multi_scale size - # opt.num_workers = 0 # bug https://github.com/ultralytics/yolov3/issues/174 + opt.num_workers = 0 # bug https://github.com/ultralytics/yolov3/issues/174 else: torch.backends.cudnn.benchmark = True # unsuitable for multiscale