Преглед изворни кода

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):
     def __call__(self, opts, **kwargs):
         opt_kwargs = self.tuner_cls.extract_kwargs(opts)
         opt_kwargs = self.tuner_cls.extract_kwargs(opts)
         _kwargs = dict(self.kwargs, **kwargs, **opt_kwargs)
         _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):
     def get(self, key, default=None):
         return self.kwargs.get(key, default)
         return self.kwargs.get(key, default)