updates

2019-03-31 20:19:15 +02:00 · 2019-03-31 20:19:15 +02:00 · b56952d707
parent 09b02d2029
commit b56952d707
1 changed files with 1 additions and 0 deletions
--- 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