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

iglob bug fix

This commit is contained in:
Glenn Jocher 2020-05-17 14:31:14 -07:00
parent bbd82bb94d
commit c94019f159
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

@ -398,7 +398,7 @@ if __name__ == '__main__':
opt.weights = last if opt.resume else opt.weights
check_git_status()
opt.cfg = list(glob.iglob('./**/' + opt.cfg, recursive=True))[0] # find file
opt.data = list(glob.iglob('./**/' + opt.data, recursive=True))[0] # find file
# opt.data = list(glob.iglob('./**/' + opt.data, recursive=True))[0] # find file
print(opt)
opt.img_size.extend([opt.img_size[-1]] * (3 - len(opt.img_size))) # extend to 3 sizes (min, max, test)
device = torch_utils.select_device(opt.device, apex=mixed_precision, batch_size=opt.batch_size)