From 61c396f282dbdd7ffba26637b25712e408d78385 Mon Sep 17 00:00:00 2001 From: igor-cheb Date: Tue, 28 Feb 2023 10:53:36 +0600 Subject: [PATCH] run_inference upd --- run_inference.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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