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

updates

This commit is contained in:
Glenn Jocher 2019-03-31 20:19:15 +02:00
parent 09b02d2029
commit b56952d707
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -29,6 +29,7 @@ def train(
if multi_scale: if multi_scale:
img_size = 608 # initiate with maximum multi_scale size img_size = 608 # initiate with maximum multi_scale size
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