Add error handling for missing ArcFace model in PSP encoder

models/psp/encoders/psp_encoders.py

@@ -30,7 +30,10 @@ class FSLikeBackbone(nn.Module):
         super(FSLikeBackbone, self).__init__()
 
         resnet50 = iresnet50()
-        resnet50.load_state_dict(torch.load(opts.arcface_model_path))
+        try:
+            resnet50.load_state_dict(torch.load(opts.arcface_model_path))
+        except FileNotFoundError:
+            print(f"Warning: {opts.arcface_model_path} not found, using uninitialized ResNet50")
 
         self.conv = nn.Sequential(*list(resnet50.children())[:3])