Commit 6031d7d7 authored by JessicaOjo's avatar JessicaOjo
Browse files

remove test limit

parent 2a285714
......@@ -9,17 +9,18 @@ task=afrimgsm_direct_eng,afrimgsm_direct_fra,afrimgsm_direct_swa #afrimgsm_direc
for model in "${models[@]}"
do
echo "Evaluating model: $model"
for fewshot in 0 2 4 6 8
do
export OUTPUT_DIR=results/$fewshot
# for fewshot in 0 2 4 6 8
# do
export OUTPUT_DIR=results/$fewshot
mkdir -p "$OUTPUT_DIR"
mkdir -p "$OUTPUT_DIR"
lm_eval --model openai-chat-completions \
--model_args model="${model}" \
--tasks $task \
--output_path "$OUTPUT_DIR" \
--num_fewshot $fewshot \
--verbosity DEBUG
done
lm_eval --model openai-chat-completions \
--model_args model="${model}" \
--tasks $task \
--output_path "$OUTPUT_DIR" \
--num_fewshot 0 \
--limit 5 \
--verbosity DEBUG
# done
done
\ No newline at end of file
......@@ -33,7 +33,6 @@ do
--batch_size 16 \
--output_path "$OUTPUT_DIR" \
--num_fewshot $fewshot \
--limit 1 \
--verbosity DEBUG
done
done
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment