model ckpt

app_utils/model_utils.py

@@ -15,7 +15,7 @@ lg.setLevel(RDLogger.CRITICAL)
 # Load model and data
 
 def load_model_components():
-    param_pth = '/data/yzhouc01/FILIP-MS/experiments/20250913_optimized_filip-model/lightning_logs/version_0/hparams.yaml'
+    param_pth = 'experiments/20250913_optimized_filip-model/lightning_logs/version_0/hparams.yaml'
     with open(param_pth) as f:
         params = yaml.load(f, Loader=yaml.FullLoader)
 
@@ -24,7 +24,7 @@ def load_model_components():
 
     # load model
 
-    checkpoint_pth = "/data/yzhouc01/FILIP-MS/experiments/20250913_optimized_filip-model/epoch=1993-train_loss=0.10.ckpt"
+    checkpoint_pth = "experiments/20250913_optimized_filip-model/epoch=1993-train_loss=0.10.ckpt"
     params['checkpoint_pth'] = checkpoint_pth
     model = get_model(params['model'], params)