updates
This commit is contained in:
parent
37a07a44a1
commit
01e11aee04
2
train.py
2
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
|
||||
|
||||
|
|
Loading…
Reference in New Issue