LlaMA_Factory实战微调VL大模型

VL微调数据格式文件样例如下

一:微调数据集准备

1、制作微调数据集

首先找到data文件夹下方的mllm_demo.json,确认微调VL模型时的数据格式模板,然后按照模板,制作对应的指令微调数据集。

2、确认微调模型认知

打开identity.json文件,修改模型初步认知,确认微调后模型的名称和开发者,如下。

3、进入dataset_info.json,添加数据集,修改文件夹名称和对应标注json文件名即可

复制代码
"****_vl_data(图片文件夹名称)": {
    "file_name": "****_vl_data.json(对应的json文件名称)",
    "formatting": "sharegpt",
    "columns": {
      "messages": "messages",
      "images": "images"
    },
    "tags": {
      "role_tag": "role",
      "content_tag": "content",
      "user_tag": "user",
      "assistant_tag": "assistant"
    }
  }

二:启动训练

1、先下载模型文件

2、web端启动命令

复制代码
llamafactory-cli webui

配置参数

启动训练后使用显存大小:50924MiB≈49.7G(这个有大佬知道正常吗,7B参数按照这个训练要这么大的显存?),共143条数据,15Epoch所用时间为:40min

训练曲线如下所示

但是感觉没怎么收敛啊,准备在加几轮训练一下

修改参数:参数部分可参照该链接,说的很详细

https://blog.csdn.net/qq_62223405/article/details/149500255?fromshare=blogdetail\&sharetype=blogdetail\&sharerId=149500255\&sharerefer=PC\&sharesource=weixin_42225889\&sharefrom=from_link

在15个epoch时模型训练稳定

中断训练,然后导出模型文件进行测试,完成训练

二:增加到一共300条数据,使用3B进行训练

配置如下

占用显存大小为:32364MiB≈31.6G

用时40分钟达到稳定,,epoch=12左右,中断测试

相关推荐
Token炼金师6 小时前
引擎四强:vLLM、SGLang、TensorRT-LLM 与 llama.cpp —— 推理引擎选型对决
人工智能·llm·llama·vllm·tensorrt-llm·sglang
尼米棕熊1 天前
大模型学习8上-推理部署框架llama.cpp与Ollama使用指北
学习·llama
Day(AKA Elin)1 天前
【Day】MTP(Multi Token Prediction)技术学习
python·深度学习·学习·llama
染指11101 天前
50.llama_index-文档分割器(文本)
llama·rag·llamaindex
Scabbards_4 天前
经典大语言模型框架整理(gpt/llama/chatglm/qwen/deepseek)
gpt·语言模型·llama
染指11105 天前
46.llama_index-提示词模板(富提示词模板、jinja2静态和动态模板)
llama·rag·llamaindex
染指11107 天前
45.llama_index-全局配置(Settings)
llama·llamaindex
俊俊谢7 天前
LLaMA-Factory 部署与 DeepSeek-R1-Distill-Qwen 模型乱码问题解决全记录
机器学习·大模型·llama·qwen·llama-factory·deepseek·hugging-face
sunywz7 天前
【AI】基于LLaMA-Factory大模型微调
人工智能·llama
AI小百科7 天前
基于 llama.cpp 的 AI 编程相关开源项目
人工智能·开源·llama