fixing pathing for stage 3 model download

RWKV · Sep 13, 2023 · 53ee8f4 · 53ee8f4
1 parent 85f2c39
commit 53ee8f4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/notebook/experiment/rwkv-x-exp/v5-r3-memory/L6-D2048-E1e-1-ctx4k/stage3.ipynb b/notebook/experiment/rwkv-x-exp/v5-r3-memory/L6-D2048-E1e-1-ctx4k/stage3.ipynb
@@ -182,7 +182,7 @@
     "        --data.sort_by_length=True \\\n",
     "        --model.ctx_len=4096 \\\n",
     "        --model.bptt_learning_range=2 \\\n",
-    "        --model.load_model=\"../model/{FILENAME_PREFIX}-mem-ctx-8k.pth\""
+    "        --model.load_model=\"../model/{FILENAME_PREFIX}-mem-ctx-512.pth\""
    ]
   },
   {

diff --git a/notebook/experiment/rwkv-x-exp/v5-r3-memory/L6-D2560-E1e-1-ctx4k/stage3.ipynb b/notebook/experiment/rwkv-x-exp/v5-r3-memory/L6-D2560-E1e-1-ctx4k/stage3.ipynb
@@ -182,7 +182,7 @@
     "        --data.sort_by_length=True \\\n",
     "        --model.ctx_len=4096 \\\n",
     "        --model.bptt_learning_range=2 \\\n",
-    "        --model.load_model=\"../model/{FILENAME_PREFIX}-mem-ctx-8k.pth\""
+    "        --model.load_model=\"../model/{FILENAME_PREFIX}-mem-ctx-512.pth\""
    ]
   },
   {