Turn opt into dictionary before sending it to yaml

2020-06-24 10:49:08 -05:00 · 2020-06-24 10:49:08 -05:00 · 611aacf1bf
parent bc4ef4861b
commit 611aacf1bf
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -193,7 +193,7 @@ def train(hyp):
        yaml.dump(hyp, f)

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