Turn opt into dictionary before sending it to yaml
Browse files
train.py
CHANGED
@@ -193,7 +193,7 @@ def train(hyp):
|
|
193 |
yaml.dump(hyp, f)
|
194 |
|
195 |
with open(os.path.join(log_dir, 'opt.yaml'), 'w') as f:
|
196 |
-
yaml.dump(opt, f)
|
197 |
|
198 |
# Class frequency
|
199 |
labels = np.concatenate(dataset.labels, 0)
|
|
|
193 |
yaml.dump(hyp, f)
|
194 |
|
195 |
with open(os.path.join(log_dir, 'opt.yaml'), 'w') as f:
|
196 |
+
yaml.dump(vars(opt), f)
|
197 |
|
198 |
# Class frequency
|
199 |
labels = np.concatenate(dataset.labels, 0)
|