diff --git a/run_inference.sh b/run_inference.sh index 48947f8d..5a2bd976 100644 --- a/run_inference.sh +++ b/run_inference.sh @@ -4,7 +4,7 @@ CUDA_VISIBLE_DEVICES=0,1 python main.py \ --user_msg rationale --img_type detr \ --bs 8 --eval_bs 4 --eval_acc 10 --output_len 512 \ --final_eval --prompt_format QCM-LE \ - --evaluate_dir models/rationale + --evaluate_dir models/MM-CoT-UnifiedQA-base-Rationale # answer inference CUDA_VISIBLE_DEVICES=0,1 python main.py \ @@ -14,4 +14,4 @@ CUDA_VISIBLE_DEVICES=0,1 python main.py \ --final_eval --prompt_format QCMG-A \ --eval_le models/rationale/predictions_ans_eval.json \ --test_le models/rationale/predictions_ans_test.json \ - --evaluate_dir models/answer \ No newline at end of file + --evaluate_dir models/MM-CoT-UnifiedQA-base-Answer \ No newline at end of file