Commit bcee8f2e authored by JessicaOjo's avatar JessicaOjo
Browse files

fix direct path, add bash script for gpt models

parent c4f634c6
...@@ -22,7 +22,7 @@ do ...@@ -22,7 +22,7 @@ do
echo "Evaluating model: $model" echo "Evaluating model: $model"
for fewshot in 0 2 4 6 8 for fewshot in 0 2 4 6 8
do do
export OUTPUT_DIR=results/$fewshot/${model##*/} export OUTPUT_DIR=results/${model##*/}/$fewshot
mkdir -p "$OUTPUT_DIR" mkdir -p "$OUTPUT_DIR"
...@@ -31,6 +31,7 @@ do ...@@ -31,6 +31,7 @@ do
--tasks $task\ --tasks $task\
--device cuda:0 \ --device cuda:0 \
--batch_size 16 \ --batch_size 16 \
--output_path "$OUTPUT_DIR" \
--num_fewshot $fewshot \ --num_fewshot $fewshot \
--verbosity DEBUG --verbosity DEBUG
done done
......
#!/bin/bash
models=(
"gpt-3.5-turbo"
"gpt-4-0125-preview"
)
task=afrimgsm_direct_amh,afrimgsm_direct_ibo,afrimgsm_direct_fra,afrimgsm_direct_sna,afrimgsm_direct_lin,afrimgsm_direct_wol,afrimgsm_direct_ewe,afrimgsm_direct_lug,afrimgsm_direct_xho,afrimgsm_direct_kin,afrimgsm_direct_twi,afrimgsm_direct_zul,afrimgsm_direct_orm,afrimgsm_direct_yor,afrimgsm_direct_hau,afrimgsm_direct_sot,afrimgsm_direct_swa
for model in "${models[@]}"
do
echo "Evaluating model: $model"
for fewshot in 0 2 4 6 8
do
export OUTPUT_DIR=results/${model##*/}/$fewshot
mkdir -p "$OUTPUT_DIR"
lm_eval --model openai-chat-completions \
--model_args model="${model}"\
--tasks $task\
--device cuda:0 \
--batch_size 16 \
--output_path "$OUTPUT_DIR" \
--num_fewshot $fewshot \
--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