diff --git a/eval_configs/minigptv2_benchmark_evaluation.yaml b/eval_configs/minigptv2_benchmark_evaluation.yaml index 8eb5ced..2efe3d9 100644 --- a/eval_configs/minigptv2_benchmark_evaluation.yaml +++ b/eval_configs/minigptv2_benchmark_evaluation.yaml @@ -51,7 +51,7 @@ evaluation_datasets: eval_file_path: /path/to/eval/annotation/path img_path: /path/to/eval/image/path max_new_tokens: 20 - batch_size: 100 + batch_size: 10 iconvqa: eval_file_path: /path/to/eval/annotation/path img_path: /path/to/eval/image/path @@ -66,7 +66,7 @@ evaluation_datasets: eval_file_path: /path/to/eval/annotation/path img_path: /path/to/eval/image/path max_new_tokens: 20 - batch_size: 100 + batch_size: 10 run: task: image_text_pretrain