swift与Internvl下的多模态大模子分布式微调指南(附代码和数据)
大模子相关目次大模子,包罗部署微调prompt/Agent应用开辟、知识库增强、数据库增强、知识图谱增强、自然语言处理、多模态等大模子应用开辟内容
从0起步,扬帆起航。
[*]基于Dify的智能分类方案:大模子结合KNN算法(附代码)
[*]OpenCompass:大模子测评工具
[*]一文读懂多模态大模子根本架构
[*]大模子管理平台:one-api使用指南
[*]大模子RAG、ROG、RCG概念科普
[*]RAGOnMedicalKG:大模子结合知识图谱的RAG实现
[*]DSPy:变革式大模子应用开辟
[*]最简明的Few-shot Prompt指南
[*]Semantic Kernel:微软大模子开辟框架——LangChain 更换
[*]对话大模子Prompt是否必要规矩点?
[*]swift与Internvl下的多模态大模子分布式微调指南(附代码和数据)
OCR多模态大模子微调全流程纪录
媒介
微调框架:swift
微调模子:internvl-chat-v1_5
微调使命:多模态大模子在指定使命上的OCR能力微调优化
微调显存:55G,多batch时对单卡要求较高(4090不能满意需求)
https://i-blog.csdnimg.cn/direct/253ad27104b7466792511f78e9f636a9.png
微调框架部署
git clone https://kkgithub.com/modelscope/swift.git
cd swift/
conda create -n swift -y python=3.11
conda activate swift
pip install -e '.'
# glm4-v还需要额外安装torchvision
pip install torchvision
# minicpm-v-v2_5-chat还需要timm
pip install timm
数据集构建及注册
下载开源数据集,以多模态OCR数据集为例:
该数据集包含2个部门,即JPG、PNG格式的带编号图像数据集和与图像数据集对应的QA问答JSONL数据集。
https://i-blog.csdnimg.cn/direct/33a7b2d2a6a0491aa074a7afe58566c1.png
https://i-blog.csdnimg.cn/direct/5b03240a96614f619a09fcd391208a43.png
探求swift在多模态使命中所需的数据格式:
https://i-blog.csdnimg.cn/direct/84111e2fa5b64b9d9964bddeb6193538.png
编写脚本,批量处理数据格式:
import json
# 写入jsonl文件
def write_jsonl(data_list, filename):
with open(filename, 'w', encoding='utf-8') as f:
for item in data_list:
json_str = json.dumps(item, ensure_ascii=False)# 将Python对象转换为JSON格式的字符串
f.write(json_str + '\n')
# 读取jsonl文件
def read_jsonl(filename):
data_list = []
with open(filename, 'r', encoding='utf-8') as f:
for line in f:
data_list.append(json.loads(line))# 将JSON格式的字符串转换为Python对象
return data_list
# 读取jsonl文件
read_data = read_jsonl('/home/super/sgq/swift/metadata.jsonl')
# 打印读取的数据
new_data = []
for item in read_data:
new_data.append({
"query": item['question'],
"response": item['answers'],
"images": ["/self_data/test/"+item['file_name']]
})
write_jsonl(new_data,'finall_data.jsonl')
为使数据能够在框架中进行应用,需进行数据集注册:
于路径:
/home/super/sgq/swift/swift/llm/data/dataset_info.json
对数据集信息进行注册,自定义数据集格式参考:
https://i-blog.csdnimg.cn/direct/2cc90e1479e24bbe80525a80a3d56482.png
需强调的是,默认数据集使用dataset_id字段进行索引定位,自定义数据集使用dataset_path即可,目前仅支持jsonl、json格式,推荐jsonl。
最后附处理后可用于微调的数据集:
链接:https://pan.baidu.com/s/1dvWs8ny2-bky_zAWIXnxrg
提取码:ybwe
微调配置
项目部署乐成,且数据集准备后,启动框架,指令如下:
cd sgq/swift/
conda activate swift
WEBUI_SERVER=0.0.0.0 swift web-ui
启动后界面如下:
https://i-blog.csdnimg.cn/direct/dfb39855ce1c4365adcfffa5bf13c6dd.png
1.选择必要微调训练的模子型号:
https://i-blog.csdnimg.cn/direct/ea5de8b1521b443aa7529b33b710dda0.png
如图所示,在选项栏中选定internvl-chat-v1_5,后方的路径和模子类型体系会自动加载。须注意的是,模子路径是项目缓存路径,由框架自动缓存并路由。未试验自定义路径时框架是否能够作用。
2.配置体系提示词和微调数据集,数据集可多选,具体如下方红色框图区域所示:
https://i-blog.csdnimg.cn/direct/4203a3f37ab844ccae4ce8d0f06a87f2.png
需注意的是,选择数据集要出现在可选下拉框中且可用,必须安装“数据集构建及注册”所强调的步骤自定义数据并注册。
3.训练方式配置
https://i-blog.csdnimg.cn/direct/a9002f05e60748c09a81ac48d4f7af55.png
默认训练方式采取LoRA,需设置精度为bf16。
https://i-blog.csdnimg.cn/direct/268a6e74bf934d3b9d214e76972ec5be.png
设置batch为1,2及以上时4090卡会报显存溢出。
学习率等参数保持不变,本文未提到的参数项均可不做改动。
Flash attention可选项勾上,这是一种显存利用优化方案。
下拉项目界面,LoRA更多参数设置如图:
https://i-blog.csdnimg.cn/direct/6469b0623b2b4edabf68c49eef0baf73.png
以上参数设置为经验之谈,网上案比方此配置结果较好。
4.点击开始训练,若无报错,表现显存加载,并最终表现预估训练时间即表现微调乐成。
https://i-blog.csdnimg.cn/direct/eef9802e1cab413fb7d86d9fa7eb036b.png
https://i-blog.csdnimg.cn/direct/7ddd7254342a481389627c98cf01734d.png
5.微调乐成后,查看训练结果(日志及模子全重)。
训练日志路径:
/home/super/output/internvl-chat-v1_5/v4-20240708-180015/runs/
训练参数:
“/home/super/output/internvl-chat-v1_5/v4-20240708-180015/training_args.json”
训练度量:
“/home/super/output/internvl-chat-v1_5/v4-20240708-180015/logging.jsonl”
https://i-blog.csdnimg.cn/direct/41f58bd62f054d9c9fc7a560c6b610ac.png
可视化脚本:
import json
loss_values = []
acc_values = []
# 打开JSONL文件
with open(r"C:\Users\12258\Desktop\logging.jsonl", 'r') as file:
for line in file:
try:
# 加载每一行的JSON对象
entry = json.loads(line)
print(entry)
# 打印出每个条目的内容
loss_values.append(entry['loss'])
acc_values.append(entry['acc'])
except:
Pass
import matplotlib.pyplot as plt
# Plot the loss values
plt.plot(loss_values, label="Loss")
plt.xlabel("Global Step")
plt.ylabel("Loss")
plt.title("Loss Over Global Steps")
plt.legend()
plt.grid(True)
plt.show()
import matplotlib.pyplot as plt
# Plot the loss values
plt.plot(acc_values, label="Acc",color='orange')
plt.xlabel("Global Step")
plt.ylabel("Acc")
plt.title("Acc Over Global Steps")
plt.legend()
plt.grid(True)
plt.show()
https://i-blog.csdnimg.cn/direct/61ea14d880cf43159f71b2c0db3ee750.png
https://i-blog.csdnimg.cn/direct/24cc99cda36e4630879bbeaa94a562bc.png
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页:
[1]