mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-05 02:20:47 +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