|
|
@@ -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
|
|
|
|