瀏覽代碼

fix: feat size was sometimes not set!

Dimitri Korsch 6 年之前
父節點
當前提交
737824a216
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      cvfinetune/finetuner/base.py

+ 2 - 0
cvfinetune/finetuner/base.py

@@ -144,6 +144,8 @@ class _ModelMixin(abc.ABC):
 			logging.info(msg.format(self.weights))
 			loader = partial(loader_func, weights=self.weights, headless=args.headless)
 
+		feat_size = self.model.meta.feature_size
+
 		if hasattr(self.clf, "output_size"):
 			feat_size = self.clf.output_size