MiniGPT-4/eval_configs/minigpt4_eval.yaml

23 lines
456 B
YAML
Raw Normal View History

2023-04-16 22:04:16 +00:00
model:
arch: mini_gpt4
2023-08-28 18:26:00 +00:00
model_type: pretrain_vicuna0
2023-04-16 22:04:16 +00:00
max_txt_len: 160
end_sym: "###"
2023-04-17 13:54:00 +00:00
low_resource: True
2023-04-16 22:04:16 +00:00
prompt_template: '###Human: {} ###Assistant: '
ckpt: 'minigpt4/output/minigpt4_stage2_finetune/20230828181/checkpoint_19.pth'
2023-04-16 22:04:16 +00:00
datasets:
cc_sbu_align:
vis_processor:
train:
name: "blip2_image_eval"
image_size: 512
2023-04-16 22:04:16 +00:00
text_processor:
train:
name: "blip_caption"
run:
task: image_text_pretrain