当前位置:首页>渔业>【VLM助力智慧渔业】AutoDL+LLama Factory+Qwen2.5-VL-3B-Instruct-AWQ 微调

【VLM助力智慧渔业】AutoDL+LLama Factory+Qwen2.5-VL-3B-Instruct-AWQ 微调

  • 2026-10-19 23:05:07
【VLM助力智慧渔业】AutoDL+LLama Factory+Qwen2.5-VL-3B-Instruct-AWQ 微调
前言
这是一个记录大模型微调的文章,旨在将VLM应用于农业,加速渔业智能化。
算力平台:https://autodl.com/home
微调框架:https://llamafactory.readthedocs.io/zh-cn/latest/
模型:https://www.modelscope.cn/models/Qwen/Qwen2.5-VL-3B-Instruct-AWQ
数据集:虾类养殖场景视觉问答,包含120张图片。见阅读原文。
正文
前置工作
  • 在AutoDL进行注册,并进行学生认证(申请学校个人邮箱)
  • 下载隧道工具:https://autodl.com/docs/ssh_proxy/
  • 租用算力,选择vGPU-32GB
  • 点击蓝色按钮,进到如下图所示页面
  • 选择镜像,如下图所示
  • 然后点击确认开机,会跳转到如下页面。
  • 点击红框内的JupyterLab进入如下页面
  • 再点击终端进入如下图所示页面
安装依赖
其中LLama-Factory和flash_attn可能因为网络问题安装失败,请重复尝试。或者本地下载后再上传至服务器(和数据集上传的方式一样)。
pip install datasets modelscopegit clone --depth 1 https://github.com/hiyouga/LLaMA-Factory.gitcd LLaMA-Factorypip install -e ".[torch,metrics]"wget https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.8cxx11abiTRUE-cp312-cp312-linux_x86_64.whlpip install flash_attn-2.8.3+cu12torch2.8cxx11abiTRUE-cp312-cp312-linux_x86_64.whlpip install autoawq --no-build-isolation
下载Qwen2.5-VL-3B-Instruct-AWQ
modelscope download --model 'Qwen/Qwen2.5-VL-3B-Instruct-AWQ' --local_dir /root/model/
出现如下界面,代表模型下载完成,连续按5次回车
上传数据集
然后输入命令解压
cd /rootunzip 20251001.zip -d /root/datasetrm 20251001.zip
输入以下命令开启LLama Factory的WEB UI界面
llamafactory-cli webui
打开下载的SSH隧道工具(双击文件夹里的exe文件),填上上图红框内的登录指令、密码、和端口
点击开启代理后等待出现如下界面,点击蓝色链接。
进入如下界面,点击红框内的按钮切换为中文
然后如下图所示调整参数,感兴趣者可以自行调节其他参数
轮流点击红框内的按钮,即可开始训练
训练过程中可以打开监控面板,入口在ssh密码同一个界面
大概训练1h,可以在webui里看到损失和训练进度
手动评估
由于数据集比较少,所以未设置验证集和测试集,可以手动测试。进入chat页面。
然后就可以开始对话了
总结
数据集大小和质量、模型参量大小、超参数都会影响最终的模型性能。
可根据具体需求选择算力。
可以仿照文中的数据集格式构建自己的数据集。
快来构建自己的模型吧!

最新文章

随机文章