LlamaFactory可视化微调大模型 - 参数详解

LlamaFactory 前言

LLaMA Factory 是一个用于微调大型语言模型的强大工具,特别是针对 LLaMA 系列模型。

可以适应不同的模型架构和大小。

支持多种微调技术,如全参数微调、LoRALow-Rank Adaptation )、QLoRAQuantized LoRA )等。

还给我们提供了简单实用的命令行接口。

支持多 cpu 训练,多任务微调,还有各种内存优化技术,如梯度检查点、梯度累积等。

支持混合精度训练,提高训练效率。

本文不再赘述 LlamaFactory 的安装过程

LlamaFactory参数基本设置

打开我们 LlamaFactoryweb 运行界面,进入根目录执行下列命令:

shell 复制代码
llamafactory-cli webui

看到下列界面 在浏览器打开我们开启的 webui 界面 http://127.0.0.1:7860 我们依次来解释每个参数的选择:

这里是语言选择

选择 zh 即可。 模型选择

选择适合自己的模型,这里都会从 Hugging Face 里面下载, 这一步是自定义路径

一般就用选择好的默认路径即可。 微调方法:

这里有三种,full全参数微调, Freeze(冻结部分参数) LoRA(Low-Rank Adaptation),还有 QLoRA 等。

全参数微调可以最大的模型适应性,可以全面调整模型以适应新任务。通常能达到最佳性能。

Freeze 训练速度比全参数微调快,会降低计算资源需求。

LoRA :显著减少了可训练参数数量,降低内存需求,训练速度快,计算效率高。还可以为不同任务保存多个小型适配器,减少了过拟合风险。

QLoRA训练速度跟 LoRA 差不多,基本保持了 LoRa 的优势,会进一步减少内存使用。

综合速度,灵活性考虑 选择 LoRA 或者 QLorRA 检查点路径:

检查点是模型训练过程中的一个快照,保存了模型的权重、优化器状态等信息。

主要用于保存训练进度允许从中断点恢复训练,性能评估等。

LlamaFactory参数高级设置

量化等级

量化等级有8位量化( INT8)和4位量化( INT4 ),QLoRA 它允许在使用低位量化(如4位)的同时,通过 LoRA 方法进行高效的微调。

量化方法

bitsandbyteshqq

Bitsandbytes:内存效率高,可以显著减少 GPU 内存使用

Hqq: 提供更多的量化选项和更细粒度的控制,使用可能稍微复杂一些,需要更多的配置。

提示模板

就是构建结构化输入的一种方式,好的提示模板可以显著提高模型的性能和适用性,为了适应不同的需求,我们要自定义模板。

RoPE 插值方法:

线性插值和动态 NTK 缩放,线性插值简单直观,动态 NTK 缩放更灵活,可以适应不同长度的输入。

加速方式

autounslothflashattn2

auto自动模式会根据你的硬件配置和当前的训练任务自动选择最适合的加速技术。这是最简单的一种方式,不需要用户进行任何额外配置。

FlashAttention2 是一种优化的注意力机制,旨在加速 Transformer 模型的训练。它通过优化内存访问和计算流程来提高训练速度。

Unsloth 是一种特定的优化技术,用于减少训练过程中的计算冗余和内存占用,从而加快训练速度。

图像输入

对于多模态模型的训练,要结合图像和文本数据进行训练和推理。 学习率

AdamW的学习率通常在1e-53e-5之间,于大型语言模型(如 BERTGPT 等)的微调,常用的学习率范围是 2e-55e-5,从一个相对较小的值开始,如 2e-5

如果训练不稳定或损失波动很大,可以尝试降低学习率,如果训练进展太慢,可以尝试略微增加学习率。 训练轮数:

对于大语言模型的微调,通常在 210epoch 之间, 轮数过多可能导致过拟合,特别是在小数据集上。

最大梯度范数

Max Gradient Norm)是一种用于防止梯度爆炸的技术,也称为梯度裁剪(Gradient Clipping)。这个参数设置了梯度的最大允许值,如果梯度超过这个值,就会被缩放到这个最大值。

通常在 0.110 之间,太小:可能会限制模型学习,太大:可能无法有效防止梯度爆炸。

最大样本数

它决定了每个数据集中使用多少样本进行训练''

如果原始数据集很大,设置一个合理的最大样本数可以减少训练时间,如果计算资源有限,较小的样本数可以加快训练速度。

计算类型

bf16 fp16 fp32 purebf16,如果你的硬件支持 bfloat16,且你希望最大化内存效率和计算速度,可以选择 bf16purebf16

如果你的硬件支持 fp16,你希望加速训练过程且能够接受较低的数值精度,可以选择 fp16

如果你不确定你的硬件支持哪些类型,或你需要高精度计算,可以选择 fp32

截断长度:

截断长度是指在处理输入序列时,模型所能接受的最大标记(token)数量。

如果输入序列超过了这个长度,多余的部分将被截断,以确保输入序列长度不会超出模型的处理能力。

对于文本分类任务,通常截断到 128256 个标记可能就足够了;而对于更复杂的任务,如文本生成或翻译,可能需要更长的长度。 批处理大小:

批处理大小是指在每次迭代中输入到模型中的样本数量。

在深度学习训练过程中,数据通常会被分成多个批次(batch)进行处理,每个批次包含一组样本。

较大的批处理大小会占用更多的内存(显存)。

如果批处理大小过大,可能导致显存不足,训练无法进行。

合理的批处理大小可以提高计算效率,大批量的数据可以更有效地利用 GPU 进行并行计算。

梯度累计:

是一种有效的策略,用于在受限的 GPU 内存情况下模拟更大的批处理大小。

验证集比例:

是指在机器学习和深度学习模型训练过程中,从训练数据集中划分出来的一部分数据,用于评估模型的性能。

验证集的数据不参与模型的训练,仅用于在训练过程中监控模型的表现,以防止过拟合和调整模型的超参数,常见的比例有 10%20% 等,具体选择取决于数据集的大小和具体的应用场景。 学习率调整:

训练过程中保持学习率不变。随着训练进行,逐步减小学习率。

每隔一定的训练轮数(epoch),将学习率按某个比例缩小。

在每个周期内,学习率呈现余弦函数形态变化。如 Adam ���AdagradRMSprop 等,根据梯度变化动态调整学习率。

训练

然后接下来选择我们微调要保存的目录即可。

然后右边会有微调过程中损失函数曲线,会越来越收敛,误差越来越小,直到某个阀值。

8.总结

LLaMA Factorywebui 里面,一切微调都是可视化,方便我们入手。

我们可以不用了解每个参数背后的具体实现手段,但一定要知道每个参数的含义,大模型的微调不是一蹴而就,而是不断更新和优化。

我们可能要调试各种参数之间的搭配,最后得到一个接近我们任务的大模型。

-- 欢迎点赞、关注、转发、收藏【我码玄黄】,gonghao同名

相关推荐
Alluxio10 分钟前
选择Alluxio来解决AI模型训练场景数据访问的五大理由
大数据·人工智能·分布式·ai·语言模型
AIPaPerPass写论文17 分钟前
写论文去哪个网站?2024最佳五款AI毕业论文学术网站
人工智能·深度学习·chatgpt·powerpoint·ai写作
唯创知音19 分钟前
电子烟智能化创新体验:WTK6900P语音交互芯片方案,融合频谱计算、精准语音识别与流畅音频播报
人工智能·单片机·物联网·音视频·智能家居·语音识别
5pace40 分钟前
PyTorch深度学习快速入门教程【土堆】基础知识篇
人工智能·pytorch·深度学习
aWty_1 小时前
机器学习--卷积神经网络(包括python实现)
人工智能·机器学习·cnn
wx7408513261 小时前
小琳AI课堂:大语言模型如何符合伦理限制
人工智能·语言模型·自然语言处理
5pace1 小时前
机器学习(西瓜书)第 14 章 概率图模型
人工智能·机器学习
Linux猿1 小时前
828华为云征文 | 云服务器Flexus X实例:部署 AgentOps,全方位监测智能体
人工智能·华为云·agent·autogen·flexus云服务器x实例·华为云服务器·agentops
Rense12 小时前
开源RK3588 AI Module7,并与Jetson Nano生态兼容的低功耗AI模块
人工智能·开源
FL16238631292 小时前
[数据集][目标检测]不同颜色的安全帽检测数据集VOC+YOLO格式7574张5类别
人工智能·yolo·目标检测