大家好,我是herosunly。985院校硕士毕业,现担任算法研究员一职,热衷于机器学习算法研究与应用。曾获得阿里云天池比赛第一名,CCF比赛第二名,科大讯飞比赛第三名。拥有多项发明专利。对机器学习和深度学习拥有自己独到的见解。曾经辅导过若干个非计算机专业的学生进入到算法行业就业。希望和大家一起成长进步。
本文主要介绍了ChatGLM ptuning 的实战方案,希望对学习大语言模型的同学们有所帮助。
文章目录
- 0. 简介
- 1. 安装虚拟环境
- 2. 下载文件
- 2.1 下载源代码
- 2.2 下载模型依赖文件
- 2.3 下载模型文件
- 3. 安装依赖库
- 4. ptuning训练
- 5. 运行微调后的ChatGLM
0. 简介
对特定数据集进行微调,具体方法包括Prompt Tuning(v1、v2)、Freeze、Lora。本次使用的是Prompt Tuning v2,所需显存比较小(7GB),具体来说