Skip to content

Latest commit

 

History

History

GLM-4-9B-CoT_LoRA

训练流程

进入本项目目录,训练启动命令如下:

python finetune.py data ../../../Model/THUDM/glm-4-9b-chat configs/lora.yaml

从保存检查点继续训练,命令如下:

python finetune.py data ../../../Model/THUDM/glm-4-9b-chat configs/lora.yaml yes

多卡微调

需要使用conda安装mpi4py

conda install mpi4py 
OMP_NUM_THREADS=1 torchrun --standalone --nnodes=1 --nproc_per_node=8  finetune.py  data  ../../../Model/THUDM/glm-4-9b-chat  configs/sft.yaml