@@ -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)
Dimitri Korsch