From b56952d70764b2dc895a6cbc4d7bf5b6cb19c743 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 31 Mar 2019 20:19:15 +0200 Subject: [PATCH] updates --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index da102e4dc2..d0fbd837f9 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