Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Turn opt into dictionary before sending it to yaml

Browse Source
This commit is contained in:
Alex Stoken committed 2020-06-24 10:49:08 -05:00
1 parent bc4ef4861b
commit 611aacf1bf
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+1 -1
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)