minchul
/

cvlface_arcface_ir101_webface4m

Feature Extraction

Model card Files Files and versions

minchul commited on Jun 7, 2024

Commit

87b3115

·

verified ·

1 Parent(s): 6dadf30

Upload model

Files changed (1) hide show

wrapper.py +2 -2

wrapper.py CHANGED Viewed

@@ -23,8 +23,8 @@ class CVLFaceRecognitionModel(PreTrainedModel):
         self.model = get_model(model_conf)
         self.model.load_state_dict_from_path('pretrained_model/model.pt')
-    def forward(self, x):
-        return self.model(x)

         self.model = get_model(model_conf)
         self.model.load_state_dict_from_path('pretrained_model/model.pt')
+    def forward(self, *args, **kwargs):
+        return self.model(*args, **kwargs)