Upload weborg_rt_d24_s3500_blue_recipe_s0/code_diff.patch with huggingface_hub
Browse files
weborg_rt_d24_s3500_blue_recipe_s0/code_diff.patch
ADDED
|
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
diff --git a/runs/weborg_redteam_d24.sh b/runs/weborg_redteam_d24.sh
|
| 2 |
+
index 2e4334f..49fd549 100755
|
| 3 |
+
--- a/runs/weborg_redteam_d24.sh
|
| 4 |
+
+++ b/runs/weborg_redteam_d24.sh
|
| 5 |
+
@@ -90,7 +90,7 @@ else
|
| 6 |
+
--carry runs/weborg_redteam/blue_recipe.json || die "deriving the arms failed"
|
| 7 |
+
fi
|
| 8 |
+
for spec in $ARMS; do
|
| 9 |
+
- [ -f "$ARM_DIR/${spec%%:*}.json" ] || die "unknown arm '${spec%%:*}' (ARM_DIR holds: $(ls "$ARM_DIR" | sed 's/\.json//' | tr '\n' ' '))"
|
| 10 |
+
+ [ -f "$ARM_DIR/${spec%%[@:]*}.json" ] || die "unknown arm '${spec%%[@:]*}' (ARM_DIR holds: $(ls "$ARM_DIR" | sed 's/\.json//' | tr '\n' ' '))"
|
| 11 |
+
done
|
| 12 |
+
|
| 13 |
+
echo "================================================================"
|
| 14 |
+
@@ -110,8 +110,9 @@ run_eval() { # <script> <log> [env...]
|
| 15 |
+
|
| 16 |
+
FAILED=()
|
| 17 |
+
for spec in $ARMS; do
|
| 18 |
+
- ARM="${spec%%:*}"; NSEEDS="${spec#*:}"; [ "$NSEEDS" = "$spec" ] && NSEEDS=1
|
| 19 |
+
- for SEED in $(seq 0 $((NSEEDS - 1))); do
|
| 20 |
+
+ ARM="${spec%%[@:]*}"
|
| 21 |
+
+ if [[ "$spec" == *@* ]]; then SEEDS="${spec#*@}"; else NSEEDS="${spec#*:}"; [ "$NSEEDS" = "$spec" ] && NSEEDS=1; SEEDS=$(seq 0 $((NSEEDS - 1))); fi
|
| 22 |
+
+ for SEED in $SEEDS; do
|
| 23 |
+
EXP="${NAME}_${ARM}_s${SEED}"
|
| 24 |
+
DIR="$WORK_DIR/experiments/$EXP"
|
| 25 |
+
export NANOCHAT_EXPERIMENT="$EXP" NANOCHAT_DATASET="$DATASET" NANOCHAT_TOKENIZER="$TOKENIZER" \
|
| 26 |
+
@@ -123,7 +124,7 @@ for spec in $ARMS; do
|
| 27 |
+
|
| 28 |
+
if done_ "$DIR/base_train.log"; then echo " base_train: done"; else
|
| 29 |
+
torchrun --standalone --nproc_per_node="$NPROC" -m scripts.base_train -- \
|
| 30 |
+
- --depth="$DEPTH" --target-param-data-ratio="$RATIO" $FP8 --seed="$SEED" --data-seed="$SEED" \
|
| 31 |
+
+ --depth="$DEPTH" --target-param-data-ratio="$RATIO" $FP8 ${EXTRA:-} --seed="$SEED" --data-seed="$SEED" \
|
| 32 |
+
--core-metric-every=-1 --sample-every=-1 --run=dummy > "$DIR/base_train.log" 2>&1
|
| 33 |
+
done_ "$DIR/base_train.log" || { echo " base_train: FAILED (see $DIR/base_train.log)"; FAILED+=("$EXP"); continue; }
|
| 34 |
+
grep -h "^summary " "$DIR/base_train.log" | grep -oE "tokens_trained=[0-9]+|train_time_sec=[0-9.]+" | tr '\n' ' '; echo
|