Преглед на файлове

fixed passing of the config (used command line arguments) to the finetuner

Dimitri Korsch преди 3 години
родител
ревизия
919f17a11e
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      cvfinetune/finetuner/factory.py

+ 1 - 1
cvfinetune/finetuner/factory.py

@@ -42,7 +42,7 @@ class FinetunerFactory(object):
     def __call__(self, opts, **kwargs):
         opt_kwargs = self.tuner_cls.extract_kwargs(opts)
         _kwargs = dict(self.kwargs, **kwargs, **opt_kwargs)
-        return self.tuner_cls(config=opt_kwargs, **_kwargs)
+        return self.tuner_cls(config=opts.__dict__, **_kwargs)
 
     def get(self, key, default=None):
         return self.kwargs.get(key, default)