Dimitri Korsch 5 anni fa
parent
commit
2636439e65
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  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)
 
 		with open(folder / "args.yml", "w") as f:
-			pyaml.dump(args.__dict__, f, sort_keys=True)
+			pyaml.dump(opts.__dict__, f, sort_keys=True)