add minigptv2_train md

This commit is contained in:
junchen14 2023-10-25 08:32:56 +03:00
parent 7073c19bb3
commit f58149df4f

View File

@ -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: