From 611aacf1bfc63a6bb556a5f27ac5e560cd8b1191 Mon Sep 17 00:00:00 2001
From: Alex Stoken <alex.stoken@gmail.com>
Date: Wed, 24 Jun 2020 10:49:08 -0500
Subject: [PATCH] Turn opt into dictionary before sending it to yaml

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index a3f43d6..27a6d9f 100644
--- 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)