2024-11-12 12:22:21 +08:00
|
|
|
|
夸克网盘 docker链接：https://pan.quark.cn/s/4cda395f13e8(没有会员请联系我下载)
|
2024-11-12 11:16:38 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1.使用llama-factory对九格模型进行全参数微调。数据集见dataset
|
|
|
|
|
|
|
|
|
|
|
|
2.训练和推理都已验证无误，在A100*8卡机器上。
|
|
|
|
|
|
docker 启动：sudo docker run -it --runtime=nvidia --gpus all --shm-size=256g wjf:train
|
|
|
|
|
|
推理：python inference.py
|
|
|
|
|
|
训练：
|
|
|
|
|
|
cd training
|
|
|
|
|
|
sh training.sh
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3.推理使用多checkpoint、多次推理融合。
|
|
|
|
|
|
|
|
|
|
|
|
4.所有资料都已打包进docker，只需要docker即可。
|
|
|
|
|
|
|
|
|
|
|
|
5.启动训练时将覆盖提交的checkpoint。
|
|
|
|
|
|
|
|
|
|
|
|
6.docker卡在数据处理可能是机器的问题，尝试docker中输入：
|
|
|
|
|
|
export NCCL_DEBUG=INFO
|
|
|
|
|
|
export NCCL_SHM_DISABLE=1
|
|
|
|
|
|
export NCCL_P2P_DISABLE=1
|
|
|
|
|
|
由于需要保存多个checkpoint，请务必保证磁盘空间足够，大于500G。
|
|
|
|
|
|
|
|
|
|
|
|
7.提交不易，请有问题是及时联系我（电话：13121813131）
|
