feat: auto select best checkpoint

domino-training-test.ipynb

@@ -1052,9 +1052,14 @@
     "        surf_factors = None  # If not available, set to None\n",
     "    \n",
     "    # Load the best model checkpoint\n",
-    "    best_checkpoint = torch.load(CHECKPOINT_DIR / f\"best_model/DoMINO.0.{CKPT_NUMBER}.pt\")\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",
+    "    )\n",
     "    model.load_state_dict(best_checkpoint)  # Load the model state\n",
-    "    print(\"Model loaded\")\n",
+    "    print(f\"Model loaded: {best_checkpoint}\")\n",
     "    \n",
     "    # Set the path to save predictions\n",
     "    pred_save_path = SAVE_PATH\n",