File tree Expand file tree Collapse file tree 1 file changed +8
-7
lines changed
Expand file tree Collapse file tree 1 file changed +8
-7
lines changed Original file line number Diff line number Diff line change 1515 BASE_MODEL=$MODEL
1616fi
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
2024echo $FILE_HEADER
2125bigcodebench.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
3334bigcodebench.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
You can’t perform that action at this time.
0 commit comments