split merge train

RWKV · Oct 9, 2023 · 662230d · 662230d
1 parent 970cd15
commit 662230d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/notebook/experiment/rwkv-x-exp/multi-size-train/v5-L6-D2048-split-train.ipynb b/notebook/experiment/rwkv-x-exp/multi-size-train/v5-L6-D2048-split-train.ipynb
@@ -252,8 +252,8 @@
    "source": [
     "# Lets export the model from the checkpoint\n",
     "!cd \"{TRAINER_DIR}\" && \\\n",
-    "    python3 export_checkpoint.py \"../checkpoint/{FILENAME_PREFIX}-baseline-p3/last.ckpt\" \"../model/{FILENAME_PREFIX}-baseline-p3.pth\" \"bf16\"\n",
-    "!cd \"{TRAINER_DIR}\" && ls -alh \"../model/{FILENAME_PREFIX}-baseline-p3.pth\""
+    "    python3 export_checkpoint.py \"../checkpoint/{FILENAME_PREFIX}-split-merge-p3/last.ckpt\" \"../model/{FILENAME_PREFIX}-split-merge-p3.pth\" \"bf16\"\n",
+    "!cd \"{TRAINER_DIR}\" && ls -alh \"../model/{FILENAME_PREFIX}-split-merge-p3.pth\""
    ]
   },
   {
@@ -264,7 +264,7 @@
    "source": [
     "# # Lets do a quick dragon prompt validation\n",
     "!cd \"{INFERENCE_DIR}\" && \\\n",
-    "    python3 dragon_test.py \"../model/{FILENAME_PREFIX}-baseline-p3.pth\" \"cuda fp32\""
+    "    python3 dragon_test.py \"../model/{FILENAME_PREFIX}-split-merge-p3.pth\" \"cuda fp32\""
    ]
   }
  ],