alexnasa commited on
Commit
bbe8523
·
verified ·
1 Parent(s): f2a06a2

Update eval/tools/face_id.py

Browse files
Files changed (1) hide show
  1. eval/tools/face_id.py +1 -0
eval/tools/face_id.py CHANGED
@@ -54,6 +54,7 @@ class FaceID:
54
 
55
  self.detector = facer.face_detector("retinaface/resnet50", device=device)
56
  face_model_path = os.getenv("FACE_ID_MODEL_PATH")
 
57
  self.model = Backbone(num_layers=50, drop_ratio=0.6, mode='ir_se')
58
  self.model.load_state_dict(torch.load(face_model_path, map_location=device))
59
  self.model.to(device)
 
54
 
55
  self.detector = facer.face_detector("retinaface/resnet50", device=device)
56
  face_model_path = os.getenv("FACE_ID_MODEL_PATH")
57
+ print(f'this is the way {face_model_path}')
58
  self.model = Backbone(num_layers=50, drop_ratio=0.6, mode='ir_se')
59
  self.model.load_state_dict(torch.load(face_model_path, map_location=device))
60
  self.model.to(device)