ShareGPT数据集用于ChatGLM微调的教程:智谱模型加载ShareGPT数据的完整流程

大枫君_4481

大枫君_4481

2026-05-20

340人浏览

原创

应将ShareGPT数据转换为ChatGLM兼容的instruction-input-output格式,或通过LLaMA-Factory直接加载,或修改ChatGLM-Efficient-Tuning源码原生支持,最后验证labels掩码仅覆盖gpt输出。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

sharegpt数据集用于chatglm微调的教程:智谱模型加载sharegpt数据的完整流程

如果您计划使用ShareGPT格式的数据集对ChatGLM系列模型(如ChatGLM3-6B)进行监督微调,但发现数据加载失败或训练中断,则很可能是由于ShareGPT结构与ChatGLM默认期望的instruction格式不兼容。以下是将ShareGPT数据适配至ChatGLM微调流程的完整操作步骤:

一、确认ShareGPT数据结构并转换为ChatGLM兼容格式

ChatGLM官方微调脚本(如train_bash.py)默认接受instruction-input-output三元组结构,而ShareGPT采用conversations数组嵌套human/gpt/function_call等角色字段,需先做结构映射。该步骤确保每条样本能被tokenizer正确截断并构造labels掩码。

1、打开ShareGPT原始JSON文件,定位到每个样本的conversations数组。

2、遍历conversations中所有from: "human"和from: "gpt"条目,按顺序拼接为单轮或多轮对话文本,保留原始语义顺序。

3、将首条human内容设为instruction字段,后续human与gpt交替视为多轮上下文,末尾gpt输出设为output;若无明确gpt结尾,则丢弃该样本。

4、生成新JSONL文件,每行一个标准ChatGLM SFT样本:{"instruction": "用户第一句话", "input": "", "output": "模型最后一句回复"}。

二、使用LLaMA-Factory统一加载ShareGPT数据

LLaMA-Factory提供内置ShareGPT数据集解析器,可跳过手动转换,直接支持conversations字段读取,并自动处理system提示与tool调用标记。该方式避免JSON结构误解析,适用于含function_call/observation字段的复杂ShareGPT样本。

1、安装LLaMA-Factory:执行git clone https://github.com/hiyouga/LLaMA-Factory.git并进入目录。

2、在data子目录下新建sharegpt_zh_custom.json,直接粘贴原始ShareGPT数据(无需修改字段名)。

3、运行微调命令时指定数据集类型:--dataset_dir ./data --dataset sharegpt_zh_custom --template sharegpt。

4、确保--template sharegpt参数启用,LLaMA-Factory将调用sharegpt.py模板完成role-to-instruction映射与tokenization对齐。

绘蛙AI高清
绘蛙AI高清

绘蛙AI高清是一款AI图片处理工具,绘蛙推出的 AI 图像高清修复工具。

下载

三、修改ChatGLM-Efficient-Tuning源码以原生支持ShareGPT

当需深度控制对话轮次截断逻辑或保留observation中间态时,可直接修改ChatGLM-Efficient-Tuning的数据加载模块。该方法绕过格式转换损耗,保留原始数据语义完整性,适用于强化学习(PPO)阶段的观测序列建模。

1、打开src/data.py,找到get_dataset函数,在elif dataset_name == "alpaca_zh"分支后新增elif dataset_name == "sharegpt_zh"分支。

2、在该分支内调用自定义解析函数load_sharegpt_dataset,逐条读取conversations并构建input_ids与labels张量,对function_call和observation字段添加特殊token标识。

3、在src/train_bash.py的参数校验处,将"sharegpt_zh"加入MODEL_CLASSES支持列表。

4、启动训练时传入--dataset sharegpt_zh --stage sft,框架将自动加载并处理原始ShareGPT结构。

四、验证数据加载是否成功

在执行正式训练前,必须验证数据已正确注入模型输入管道。该步骤通过打印batch样本来确认instruction-output对齐、token长度未超限、labels掩码仅覆盖gpt输出部分。

1、在src/train_bash.py中找到trainer.train()调用前的位置,插入调试代码:print('Sample input:', tokenizer.decode(train_dataset[0]["input_ids"][:50]))。

2、运行训练脚本并添加--do_train False --max_steps 1参数,强制只加载一个batch后退出。

3、检查日志中输出的decoded文本:确认human内容出现在开头,gpt内容紧随其后,且function_call字段被转为可识别token而非乱码。

4、重点核对labels张量:其中-100应覆盖所有非gpt位置,仅gpt输出对应位置为真实token id,任何非-100值出现在human段即表示掩码错误。

相关专题

更多
C语言变量命名
C语言变量命名

c语言变量名规则是:1、变量名以英文字母开头;2、变量名中的字母是区分大小写的;3、变量名不能是关键字;4、变量名中不能包含空格、标点符号和类型说明符。php中文网还提供c语言变量的相关下载、相关课程等内容,供大家免费下载使用。

2023.06.20

3789

3

c语言入门自学零基础
c语言入门自学零基础

C语言是当代人学习及生活中的必备基础知识,应用十分广泛,本专题为大家c语言入门自学零基础的相关文章,以及相关课程,感兴趣的朋友千万不要错过了。

2023.07.25

2488

9

c语言运算符的优先级顺序
c语言运算符的优先级顺序

c语言运算符的优先级顺序是括号运算符 > 一元运算符 > 算术运算符 > 移位运算符 > 关系运算符 > 位运算符 > 逻辑运算符 > 赋值运算符 > 逗号运算符。本专题为大家提供c语言运算符相关的各种文章、以及下载和课程。

2023.08.02

1440

5

c语言数据结构
c语言数据结构

数据结构是指将数据按照一定的方式组织和存储的方法。它是计算机科学中的重要概念,用来描述和解决实际问题中的数据组织和处理问题。数据结构可以分为线性结构和非线性结构。线性结构包括数组、链表、堆栈和队列等,而非线性结构包括树和图等。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.09

1378

4

c语言random函数用法
c语言random函数用法

c语言random函数用法:1、random.random,随机生成(0,1)之间的浮点数;2、random.randint,随机生成在范围之内的整数,两个参数分别表示上限和下限;3、random.randrange,在指定范围内,按指定基数递增的集合中获得一个随机数;4、random.choice,从序列中随机抽选一个数;5、random.shuffle,随机排序。

2023.09.05

1456

5

c语言const用法
c语言const用法

const是关键字,可以用于声明常量、函数参数中的const修饰符、const修饰函数返回值、const修饰指针。详细介绍:1、声明常量,const关键字可用于声明常量,常量的值在程序运行期间不可修改,常量可以是基本数据类型,如整数、浮点数、字符等,也可是自定义的数据类型;2、函数参数中的const修饰符,const关键字可用于函数的参数中,表示该参数在函数内部不可修改等等。

2023.09.20

2438

7

c语言get函数的用法
c语言get函数的用法

get函数是一个用于从输入流中获取字符的函数。可以从键盘、文件或其他输入设备中读取字符,并将其存储在指定的变量中。本文介绍了get函数的用法以及一些相关的注意事项。希望这篇文章能够帮助你更好地理解和使用get函数 。

2023.09.20

4060

8

c数组初始化的方法
c数组初始化的方法

c语言数组初始化的方法有直接赋值法、不完全初始化法、省略数组长度法和二维数组初始化法。详细介绍:1、直接赋值法,这种方法可以直接将数组的值进行初始化;2、不完全初始化法,。这种方法可以在一定程度上节省内存空间;3、省略数组长度法,这种方法可以让编译器自动计算数组的长度;4、二维数组初始化法等等。

2023.09.22

19375

6

c语言中null和NULL的区别
c语言中null和NULL的区别

c语言中null和NULL的区别是:null是C语言中的一个宏定义,通常用来表示一个空指针,可以用于初始化指针变量,或者在条件语句中判断指针是否为空;NULL是C语言中的一个预定义常量,通常用来表示一个空值,用于表示一个空的指针、空的指针数组或者空的结构体指针。

2023.09.22

569

3

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.7万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 134.3万人学习