EPFL-MLO commited on
Commit
9b085a6
·
verified ·
1 Parent(s): 2be502d

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