chen459664's picture
Add files using upload-large-folder tool
0ecf59c verified
Raw
History Blame Contribute Delete
309 Bytes
#!/bin/bash
configure_id=$1
model=$2
# output_dir=$3
task=$3
dir=$4
# task=hellaswag,rte,mmlu
echo "${configure_id} ${task}"
lm_eval --model hf \
--model_args pretrained=${model} \
--tasks $task \
--device cuda:0 \
--batch_size 16 \
--output_path results/${dir}/${configure_id}.json