From 428deb973afbe8d5a279bcc952e9f15d21270db3 Mon Sep 17 00:00:00 2001 From: Kirill Semin Date: Sat, 17 Feb 2024 14:57:09 +0300 Subject: [PATCH] Edited script a little --- run_models.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run_models.sh b/run_models.sh index 611e718253..63fb76a0da 100644 --- a/run_models.sh +++ b/run_models.sh @@ -8,7 +8,7 @@ model_args_list=( # Iterate over the list and run the command for model_args_value in "${model_args_list[@]}"; do - command="accelerate launch -m lm_eval --model hf --tasks winogrande_ru --batch_size 32 --model_args $model_args_value" + command="accelerate launch -m lm_eval --model hf --tasks winogrande_ru,arc_challenge_ru,hellaswag_ru,mmlu_ru,gsm8k_ru,truthfulqa_mc2_ru --batch_size 4 --model_args $model_args_value" echo "Running command: $command" $command done