Skip to content

Commit a576763

Browse files
committed
fix: change split and subset args
1 parent 677a130 commit a576763

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

run.sh

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,25 +15,26 @@ else
1515
BASE_MODEL=$MODEL
1616
fi
1717

18-
FILE_HEADER=$ORG$BASE_MODEL--$DATASET-$SUBSET--$BACKEND-$TEMP-$N_SAMPLES
18+
if [ "$SUBSET" = "full" ]; then
19+
FILE_HEADER="${ORG}${BASE_MODEL}--${DATASET}-${SPLIT}--${BACKEND}-${TEMP}-${N_SAMPLES}"
20+
else
21+
FILE_HEADER="${ORG}${BASE_MODEL}--${DATASET}-${SUBSET}-${SPLIT}--${BACKEND}-${TEMP}-${N_SAMPLES}"
22+
fi
1923

2024
echo $FILE_HEADER
2125
bigcodebench.generate \
2226
--tp $NUM_GPU \
2327
--model $MODEL \
24-
--bs $BS \
25-
--temperature $TEMP \
26-
--n_samples $N_SAMPLES \
2728
--resume \
2829
--split $SPLIT \
2930
--subset $SUBSET \
3031
--backend $BACKEND \
31-
--trust_remote_code
32+
--greedy
3233

3334
bigcodebench.sanitize --samples $FILE_HEADER.jsonl --calibrate
3435

3536
# Check if the ground truth works on your machine
36-
bigcodebench.evaluate --split $SUBSET --samples $FILE_HEADER-sanitized-calibrated.jsonl
37+
bigcodebench.evaluate --split $SPLIT --subset $SUBSET --samples $FILE_HEADER-sanitized-calibrated.jsonl
3738

3839
# If the execution is slow:
39-
bigcodebench.evaluate --split $SUBSET --samples $FILE_HEADER-sanitized-calibrated.jsonl --parallel 32
40+
bigcodebench.evaluate --split $SPLIT --subset $SUBSET --samples $FILE_HEADER-sanitized-calibrated.jsonl --parallel 32

0 commit comments

Comments
 (0)