updated hf model

nasa-nccs-hpda · Jun 21, 2024 · 8b6fee0 · 8b6fee0
1 parent 970706f
commit 8b6fee0
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/notebooks/satvision-toa-reconstruction.ipynb b/notebooks/satvision-toa-reconstruction.ipynb
@@ -86,13 +86,15 @@
    "source": [
     "### Clone model ckpt from huggingface\n",
     "\n",
+    "### Model repo: https://huggingface.co/nasa-cisto-data-science-group/satvision-toa-huge-patch8-window8-128\n",
+    "\n",
     "```bash\n",
     "# On prism/explore\n",
     "module load git-lfs\n",
     "\n",
     "git lfs install\n",
     "\n",
-    "git clone git clone git@hf.co:nasa-cisto-data-science-group/satvision-toa-huge-patch8-window12-192\n",
+    "git clone git clone git@hf.co:nasa-cisto-data-science-group/satvision-toa-huge-patch8-window8-128\n",
     "```\n",
     "\n",
     "Note: If using git w/ ssh, make sure you have ssh keys enabled to clone using ssh auth.\n",
@@ -123,8 +125,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "MODEL_PATH: str = '../../satvision-toa-huge-patch8-window12-192/mp_rank_00_model_states.pt'\n",
-    "CONFIG_PATH: str = '../../satvision-toa-huge-patch8-window12-192/mim_pretrain_swinv2_satvision_huge_192_window12_100ep.yaml'\n",
+    "MODEL_PATH: str = '../../satvision-toa-huge-patch8-window8-128/mp_rank_00_model_states.pt'\n",
+    "CONFIG_PATH: str = '../../satvision-toa-huge-patch8-window8-128/mim_pretrain_swinv2_satvision_huge_128_window8_patch8_100ep.yaml'\n",
     "\n",
     "OUTPUT: str = '.'\n",
     "TAG: str = 'satvision-huge-toa-reconstruction'\n",
@@ -419,7 +421,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "pdf_path = '../../satvision-toa-reconstruction-pdf-huge-patch-8-04.30.pdf'\n",
+    "pdf_path = '../../satvision-toa-reconstruction-pdf-huge-patch-8-06.21.pdf'\n",
     "rgb_index = [0, 2, 1] # Indices of [Red band, Blue band, Green band]\n",
     "\n",
     "plot_export_pdf(pdf_path, inputs, outputs, masks, rgb_index)"