Turn opt into dictionary before sending it to yaml · 611aacf1bf - yolov5 - 拓恒技术版本管理系统

Turn opt into dictionary before sending it to yaml

5.0

Alex Stoken 4 years ago

parent

bc4ef4861b

commit

611aacf1bf

1 changed files with 1 additions and 1 deletions

Split View

							
								+ 1
								
								- 1
							
train.pyView File
								
			@@ -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)