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.
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"
ckpt: "/path/to/pretrained_checkpoint"
ckpt save path: "/path/to/save_checkpoint"
For ckpt, you may load from our pretrained model checkpoints: