astoken commited on
Commit
611aacf
·
1 Parent(s): bc4ef48

Turn opt into dictionary before sending it to yaml

Browse files
Files changed (1) hide show
  1. train.py +1 -1
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)