Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-06-12 11:30:37 +02:00
parent 37a07a44a1
commit 01e11aee04
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -76,7 +76,7 @@ def train(
if multi_scale: if multi_scale:
img_size = round((img_size / 32) * 1.5) * 32 # initiate with maximum multi_scale size 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: else:
torch.backends.cudnn.benchmark = True # unsuitable for multiscale torch.backends.cudnn.benchmark = True # unsuitable for multiscale