Fix relative paths for PyTorch models

sklearn_submission.py

@@ -200,7 +200,9 @@ def _get_dgcnn_edge_model():
         device = "cuda" if _torch.cuda.is_available() else "cpu"
     except Exception:
         device = "cpu"
-    _DGCNN_EDGE_MODEL = load_edge_model("edge_model_dgcnn.pt", device=device)
+    import os
+    model_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "edge_model_dgcnn.pt")
+    _DGCNN_EDGE_MODEL = load_edge_model(model_path, device=device)
     return _DGCNN_EDGE_MODEL
 
 
@@ -223,7 +225,9 @@ def _get_dgcnn_vertex_model():
         device = "cuda" if _torch.cuda.is_available() else "cpu"
     except Exception:
         device = "cpu"
-    _DGCNN_VERTEX_MODEL = load_vertex_model("vertex_model_dgcnn.pt", device=device)
+    import os
+    model_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "vertex_model_dgcnn.pt")
+    _DGCNN_VERTEX_MODEL = load_vertex_model(model_path, device=device)
     return _DGCNN_VERTEX_MODEL
 
 # v7: ensemble with the standalone tracks-based predictor.