zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Turn opt into dictionary before sending it to yaml

This commit is contained in:
Alex Stoken 2020-06-24 10:49:08 -05:00
parent bc4ef4861b
commit 611aacf1bf
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

@ -193,7 +193,7 @@ def train(hyp):
yaml.dump(hyp, f) yaml.dump(hyp, f)
with open(os.path.join(log_dir, 'opt.yaml'), 'w') as f: with open(os.path.join(log_dir, 'opt.yaml'), 'w') as f:
yaml.dump(opt, f) yaml.dump(vars(opt), f)
# Class frequency # Class frequency
labels = np.concatenate(dataset.labels, 0) labels = np.concatenate(dataset.labels, 0)