Update test.py

xiaoman-zhang · Mar 21, 2024 · 1678ec2 · 1678ec2
1 parent 381851e
commit 1678ec2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/MedVInT_TD/test.py b/src/MedVInT_TD/test.py
@@ -18,7 +18,7 @@
 @dataclass
 class ModelArguments:
     model_path: Optional[str] = field(default="./LLAMA/llama-7b-hf")
-    ckp: Optional[str] = field(default="./Results/QA_PMC_LLaMA_lora_PMC-CLIP_MLP/choice_training/checkpoint-4146")
+    ckp: Optional[str] = field(default="./Results/VQA_lora_PMC_LLaMA_PMCCLIP/choice/checkpoint-4000")
     checkpointing: Optional[bool] = field(default=False)
     ## Q_former ##
     N: Optional[int] = field(default=12)