Explorar o código

minor bug fixes

Dimitri Korsch %!s(int64=5) %!d(string=hai) anos
pai
achega
2636439e65
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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)