minchul
/

cvlface_adaface_ir50_ms1mv2

Feature Extraction

Model card Files Files and versions Community

minchul commited on Jun 7, 2024

Commit

d957e33

·

verified ·

1 Parent(s): f7503ec

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)