diff --git a/train.py b/train.py index da102e4d..d0fbd837 100644 --- a/train.py +++ b/train.py @@ -29,6 +29,7 @@ def train( if multi_scale: img_size = 608 # initiate with maximum multi_scale size + num_workers = 0 # bug https://github.com/ultralytics/yolov3/issues/174 else: torch.backends.cudnn.benchmark = True # unsuitable for multiscale