mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-05 18:40:46 +00:00
add minigptv2_train md
This commit is contained in:
parent
7073c19bb3
commit
f58149df4f
@ -4,9 +4,12 @@
|
|||||||
You firstly need to prepare the dataset. you can follow this step to prepare the dataset.
|
You firstly need to prepare the dataset. you can follow this step to prepare the dataset.
|
||||||
our [dataset preparation](dataset/README_MINIGPTv2_FINETUNE.md).
|
our [dataset preparation](dataset/README_MINIGPTv2_FINETUNE.md).
|
||||||
|
|
||||||
in train_configs/minigptv2_finetune.yaml, you need to set up the paths
|
In the train_configs/minigptv2_finetune.yaml, you need to set up the paths:
|
||||||
|
|
||||||
llama_model checkpoint path: "/path/to/llama_checkpoint"
|
llama_model checkpoint path: "/path/to/llama_checkpoint"
|
||||||
|
|
||||||
ckpt: "/path/to/pretrained_checkpoint"
|
ckpt: "/path/to/pretrained_checkpoint"
|
||||||
|
|
||||||
ckpt save path: "/path/to/save_checkpoint"
|
ckpt save path: "/path/to/save_checkpoint"
|
||||||
|
|
||||||
For ckpt, you may load from our pretrained model checkpoints:
|
For ckpt, you may load from our pretrained model checkpoints:
|
||||||
|
Loading…
Reference in New Issue
Block a user