| case $1 in | |
| gptj) | |
| MODEL=EleutherAI/gpt-j-6B | |
| ;; | |
| mgpt) | |
| MODEL=sberbank-ai/mGPT | |
| ;; | |
| *) | |
| MODEL=$1 | |
| esac | |
| task_no_slash=$(basename "$2") | |
| \time -o ../outputs/time-$1.$task_no_slash.txt -f "%E" python3 -m main \ | |
| --model_api_name 'hf-causal' \ | |
| --model_args pretrained=$MODEL \ | |
| --task_name $2 \ | |
| --num_fewshot 0 \ | |
| --use_cache \ | |
| --device cuda \ | |
| --batch_size $3 | |