Dimitri Korsch 4 жил өмнө
parent
commit
2636439e65

+ 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)