script stage1.sh
Browse files- scripts_full/stage1.sh +9 -0
scripts_full/stage1.sh
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
# stage1: full-data SFT (28L) + eval on the arena's real partitions
|
| 3 |
+
set -e
|
| 4 |
+
cd $HOME/mt92
|
| 5 |
+
export TOKENIZERS_PARALLELISM=false
|
| 6 |
+
echo "##### f28: SFT from base on ex/full.jsonl"
|
| 7 |
+
[ -d f28 ] || OUT=f28 EPOCHS=4 LR=8e-5 TRAIN=ex/full.jsonl ./venv/bin/python sft.py 2>&1 | tr '\r' '\n' | grep -E "train_loss|saved|Error|error"
|
| 8 |
+
for s in val_all test_all gen_holdout; do ./venv/bin/python evalset.py f28 ex/$s.jsonl; done
|
| 9 |
+
echo STAGE1_DONE
|