소스 검색

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

Dimitri Korsch 4 년 전
부모
커밋
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)