瀏覽代碼

minor bug fixes

Dimitri Korsch 5 年之前
父節點
當前提交
2636439e65
共有 1 個文件被更改,包括 1 次插入 和 1 次删除
  1. 1 1
      cvfinetune/finetuner/base.py

+ 1 - 1
cvfinetune/finetuner/base.py

@@ -340,7 +340,7 @@ class _TrainerMixin(abc.ABC):
 		folder.mkdir(parents=True, exist_ok=True)
 		folder.mkdir(parents=True, exist_ok=True)
 
 
 		with open(folder / "args.yml", "w") as f:
 		with open(folder / "args.yml", "w") as f:
-			pyaml.dump(args.__dict__, f, sort_keys=True)
+			pyaml.dump(opts.__dict__, f, sort_keys=True)