mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-05 18:40:46 +00:00
add eval cfg for demo
This commit is contained in:
parent
ae564bca7b
commit
1bb5268a5b
25
eval_configs/minigptv2_eval_textvqa.yaml
Normal file
25
eval_configs/minigptv2_eval_textvqa.yaml
Normal file
@ -0,0 +1,25 @@
|
|||||||
|
model:
|
||||||
|
arch: minigpt_v2
|
||||||
|
model_type: pretrain
|
||||||
|
max_txt_len: 500
|
||||||
|
end_sym: "</s>"
|
||||||
|
low_resource: True
|
||||||
|
prompt_template: '[INST] {} [/INST]'
|
||||||
|
# ckpt: "./minigpt4/textvqa_outputs/20250112105/checkpoint_9.pth"
|
||||||
|
ckpt: "./ckpt/checkpoint_stage3.pth"
|
||||||
|
lora_r: 64
|
||||||
|
lora_alpha: 16
|
||||||
|
|
||||||
|
|
||||||
|
datasets:
|
||||||
|
cc_sbu_align:
|
||||||
|
vis_processor:
|
||||||
|
train:
|
||||||
|
name: "blip2_image_eval"
|
||||||
|
image_size: 448
|
||||||
|
text_processor:
|
||||||
|
train:
|
||||||
|
name: "blip_caption"
|
||||||
|
|
||||||
|
run:
|
||||||
|
task: image_text_pretrain
|
Loading…
Reference in New Issue
Block a user