跳到主要内容

训练引擎

训练引擎是一款无代码微调工具,它能为您完成文本处理、大语言模型(LLM)、文生图(DreamBooth)等多个领域的微调训练任务。

适用人群

如果您不想在训练模型的技术细节上花费大量时间,训练引擎是您很好的选择,因为它基本满足当前热门、最先进开源模型的微调工作。

通过简单的 UI 交互,为您在训练模型的技术细节上节省时间。训练引擎的目标就是让所有的用户都可以简单轻松地进行模型训练,而不只是科学家或者AI方面的工程师。

如何部署训练引擎

1.创建引擎

您可以这样部署您的训练引擎:

  • 创建应用,选择 Docker 作为您的应用 SDK ,同时模版选择 AutoTrain,同时你需要填写好您的密钥 (点击此处创建密钥),建议把应用设置为私密。

alt text

2.更新引擎(可选)

创建训练引擎后,如果需要更新密钥或者遇到错误,可以进入应用的设置页面选择“功能”选项选择其中的出厂重启,您的训练程序就会被重置。

alt text

等待构建和启动完成,你就可以看到训练的界面了。

alt text

3.开始训练

以 bert-base-chinese 为基础,训练一个简单的工作任务分类模型:

  • 首先在页面上选择任务为 文本分类 ,选择基础模型为 bert-base-chinese

alt text

  • 按照格式要求,上传准备好的数据

img.png

  • 点击开启训练按钮,即可开始微调任务。

您可以通过观察页面下方训练日志了解训练进度。

alt text

训练结束后,模型会自动上传到您的账号下,可以在个人主页中查看您的模型。

img.png

您可以在 模型挂件 上进行体验,或者根据你的需要部署到 模型引擎 中,对外提供服务。