模型微调测试
基础设施配置
使用云计算平台
使用vscode进行配置
打开系统盘文件夹
llamafactory基础配置
powershell
git clone --depth 1 https://github.com/hiyouga/LLaMA-Factory.git
root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# cd LLaMA-Factory/
CITATION.cff MANIFEST.in README.md assets docker examples requirements.txt setup.py tests
LICENSE Makefile README_zh.md data evaluation pyproject.toml scripts src
root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp/LLaMA-Factory
# - 创建虚拟环境对应目录,数据盘可以被保留
mkdir -p /root/autodl-tmp/conda/pkgs
conda config --add pkgs_dirs /root/autodl-tmp/conda/pkgs
mkdir -p /root/autodl-tmp/conda/envs
conda config --add envs_dirs /root/autodl-tmp/conda/envs
# - 创建 conda 虚拟环境(一定要 3.10 的 python 版本,不然和 LLaMA-Factory 不兼容)
conda create -n llama-factory python=3.10
root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp/LLaMA-Factory# conda env list
# conda environments:
#
llama-factory /root/autodl-tmp/conda/envs/llama-factory
base /root/miniconda3
root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp/LLaMA-Factory#
conda init
source /root/.bashrc
conda activate llama-factory
pip install -e ".[torch,metrics]"
(llama-factory) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp/LLaMA-Factory# llamafactory-cli version
----------------------------------------------------------
| Welcome to LLaMA Factory, version 0.9.2.dev0 |
| |
| Project page: https://github.com/hiyouga/LLaMA-Factory |
----------------------------------------------------------
(llama-factory) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp/LLaMA-Factory#
llamafactory-cli webui

也没端口转发就这样打开了,这个vscode还挺方便,我也不知道啥原理
再开一个终端
powershell
(base) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# conda activate llama-factoy ry
(llama-factory) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp#
(llama-factory) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# mkdir Hugging-Face
(llama-factory) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# export HF_ENDPOINT=https://hf-mirror.com
(llama-factory) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# export HF_HOME=/root/autodl-tmp/Hugging-Face
(llama-factory) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp#
(llama-factory) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# pip install -U huggingface_hub
(llama-factory) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# huggingface-cli download --resume-download deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B
下载好大模型
(base) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# du -sh Hugging-Face/
3.4G Hugging-Face/
(base) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# pwd
/root/autodl-tmp
(base) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# ls
Hugging-Face LLaMA-Factory conda
(base) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp#
聊天测试


编写数据集
数据集定义
llama_factory - data - README_zh.md
训练模型

学习率:开车的油门,一脚油门会过终点,太少就很慢才达到终点(最优解)
训练轮数:太少会欠拟合(没有学到足够知识)太多会过拟合模型泛化能力下降,在训练数据表现很好但是新数据会很差
最大梯度范数:当梯度的值超过这个范围时会被截断,防止梯度爆炸现象 当成保险丝
最大样本数:每轮训练中最多使用的样本数 不设置,则拿整个数据集都去训练,但是十几万的时候就会特别多
计算类型:常见float32 float16 用float16会减少内存占用会速度快,但导致精度损失
截断长度:处理长文本时如果太长超过这个阈值的部分会被截断掉,避免内存溢出
批处理大小:由于内存限制,分批次处理
梯度累计:默认情况下模型会在每个 batch 处理完后进行一次更新一个参数,但你可以通过设置这个梯度累计,让他直到处理完多个小批次的数据后才进行一次更新
验证集比例:训练时数据会被分成训练集和验证集,80与20
学习率调节器:动态调节率,列如开车一开始加速,快到地方则降速

这些参数会导致过拟合

损失从2.7降低为0
损失曲线:降低太慢就加大学习率,如果学习结束还是下降趋势,还没到底就为欠拟合。降低太快降低学习率
检查点路径:保存的是模型在训练过程中的一个中间状态,包含了模型权重、训练过程中使用的配置(如学习率、批次大小)等信息,对LoRA来说,检查点包含了训练得到的 B 和 A 这两个低秩矩阵的权重

- 若微调效果不理想,你可以:
- 使用更强的预训练模型
- 增加数据量
- 优化数据质量(数据清洗、数据增强等,可学习相关论文如何实现)
- 调整训练参数,如学习率、训练轮数、优化器、批次大小等等
导出合并后的模型
- 为什么要合并:因为 LoRA 只是通过低秩矩阵 调整原始模型的部分权重,而不直接修改原模型的权重。合并步骤将 LoRA 权重与原始模型权重融合生成一个完整的模型
- 先创建目录,用于存放导出后的模型
powershell
(base) root@autodl-container-10a44fbcf4-b07c334b:~# cd /root/autodl-tmp/
(base) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# mkdir -p Models/deepseek-r1-1.5b-merged
(base) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp#

部署模型
powershell
conda create -n fastApi python=3.10
(base) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp/Models/deepseek-r1-1.5b-merged# conda activate fastApi
(fastApi) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp/Models/deepseek-r1-1.5b-merged#
conda install -c conda-forge fastapi uvicorn transformers pytorch
pip install safetensors sentencepiece protobuf
(fastApi) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# mkdir App
(fastApi) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp# cd App/
(fastApi) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp/App# touch main.py
(fastApi) root@autodl-container-10a44fbcf4-b07c334b:~/autodl-tmp/App#
python
from fastapi import FastAPI
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
app = FastAPI()
# 模型路径
model_path = "/root/autodl-tmp/Models/deepseek-r1-1.5b-merged"
# 加载 tokenizer (分词器)
tokenizer = AutoTokenizer.from_pretrained(model_path)
### 分词器十分重要自动加载模型匹配分词器
# 加载模型并移动到可用设备(GPU/CPU)
device = "cuda" if torch.cuda.is_available() else "cpu"
model = AutoModelForCausalLM.from_pretrained(model_path).to(device)
@app.get("/generate")
async def generate_text(prompt: str):
# 使用 tokenizer 编码输入的 prompt
inputs = tokenizer(prompt, return_tensors="pt").to(device)
# 使用模型生成文本
outputs = model.generate(inputs["input_ids"], max_length=150)
# 解码生成的输出
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
return {"generated_text": generated_text}
python -m uvicorn main:app --reload --host 0.0.0.0

训练的过拟合了
python3 /root/llama.cpp/convert_hf_to_gguf.py /mnt/c/Users/DK/Desktop/投喂的资料/deepseek-r1-1.5b-merged --outfile /root/test.gguf --outtype f16
powershell
modelfile文件已经被llamafactory定义
PS C:\Users\DK\Desktop\投喂的资料\deepseek-r1-1.5b-merged> ollama create haimianbb
gathering model components
copying file sha256:a998574673a76c152a2aabce1ab7eaf0c307990dc6c0badd964617afb9d79652 100%
copying file sha256:8dd737d110cceb8396782d1b0c9196655a4a78fee4de35f056b22d1f9e96bf96 100%
copying file sha256:59cda48bbe8bab9d61ffb410e6e3c07b6d98bff73cee7c88ff8b51f95f21ab1c 100%
copying file sha256:e20ddafc659ba90242154b55275402edeca0715e5dbb30f56815a4ce081f4893 100%
copying file sha256:6ce236e90057bbc36feee740e52666fc58103659d82063fc5ddb1355551e8148 100%
copying file sha256:b0e8dce267611e5e03ebf2cd16a8c3821bbfcad441415dd6875173788f518a56 100%
converting model
creating new layer sha256:ac67fabf626c9acf8fbfbc2b6391a8e1d9d1f4ed7f93e340627df1eb421ba7df
creating new layer sha256:741e943dbd4c7642ec2e10b275b4cc1a154d97550dce8a6288e1edf56bec5e8f
creating new layer sha256:b2ad9c47ff5fee622d61048a4fe3ba330b91d26bc04578fad9d10d143dc86322
writing manifest
success
PS C:\Users\DK\Desktop\投喂的资料\deepseek-r1-1.5b-merged>

引用B站堂吉诃德拉曼查的英豪