fix: checkpoint path print

domino-training-test.ipynb

@@ -1052,14 +1052,13 @@
     "        surf_factors = None  # If not available, set to None\n",
     "    \n",
     "    # Load the best model checkpoint\n",
-    "    best_checkpoint = torch.load(\n",
-    "        max(\n",
-    "            (CHECKPOINT_DIR / \"best_model\").glob(\"DoMINO.0.*.pt\"),\n",
-    "            key=lambda p: p.stat().st_mtime,\n",
-    "        )\n",
+    "    checkpoint_path = max(\n",
+    "        (CHECKPOINT_DIR / \"best_model\").glob(\"DoMINO.0.*.pt\"),\n",
+    "        key=lambda p: p.stat().st_mtime,\n",
     "    )\n",
+    "    print(f\"Loading checkpoint: {checkpoint_path.name}\")  # Print only the checkpoint name\n",
+    "    best_checkpoint = torch.load(checkpoint_path)\n",
     "    model.load_state_dict(best_checkpoint)  # Load the model state\n",
-    "    print(f\"Model loaded: {best_checkpoint}\")\n",
     "    \n",
     "    # Set the path to save predictions\n",
     "    pred_save_path = SAVE_PATH\n",