diff --git a/docs/engines/autotrain-engine.md b/docs/engines/autotrain-engine.md index a80e32f762dfcbf657322bf535d0212a22ed3c43..bc2dcdd18c4a967e3293fe35ee24302de1ebee17 100644 --- a/docs/engines/autotrain-engine.md +++ b/docs/engines/autotrain-engine.md @@ -1,12 +1,12 @@ # 训练引擎 -训练引擎是一款无代码工具，它能为您处理自然语言（NLP）、视觉（CV）、语音等多个领域的训练任务。它是基于应用引擎搭建的，这样您就可以简单地进行部署使用。 +训练引擎是一款无代码微调工具，它能为您完成文本处理、大语言模型（LLM）、文生图（DreamBooth）等多个领域的微调训练任务。 ## 适用人群 -如果您不想在训练模型的技术细节上花费大量时间，训练引擎是您很好的选择，因为它基本适用于需要进行自然语言（NLP）、视觉（CV）、语音或表格训练的所有先进模型。 +如果您不想在训练模型的技术细节上花费大量时间，训练引擎是您很好的选择，因为它基本满足当前热门、最先进开源模型的微调工作。 -同样训练引擎也支持您使用自定义的数据集进行模型训练，这样您就可以在训练模型的技术细节上节省时间。因为训练引擎的目标就是让所有的用户都可以简单轻松地进行模型训练，而不只是科学家或者AI方面的工程师，还包括非技术人员。 +通过简单的 UI 交互，为您在训练模型的技术细节上节省时间。训练引擎的目标就是让所有的用户都可以简单轻松地进行模型训练，而不只是科学家或者AI方面的工程师。 ## 如何部署训练引擎 @@ -14,26 +14,41 @@ 您可以这样部署您的训练引擎： -- [创建应用](/apps/overview#创建应用)，选择 Docker 作为您的应用 SDK ，同时模版选择 AutoTrain，同时你需要填写好您的密钥，以及把应用设置为私密。 +- [创建应用](/apps/overview#创建应用)，选择 Docker 作为您的应用 SDK ，同时模版选择 + AutoTrain，同时你需要填写好您的密钥 ([点击此处创建密钥](https://gitee.com/profile/personal_access_tokens))，建议把应用设置为私密。 ![alt text](/img/engines/autotrain-engine/image-5.png) -### 2.更新引擎 +### 2.更新引擎（可选） -使用之前 AutoTrain 的 AutoTrain Advanced 需要您进行更新，您只需进入应用的设置页面选择“功能”选项选择其中的出厂重启，即可把 AutoTrain 的 AutoTrain Advanced 更新到最新。这样训练时，您就能让引擎更好地适配您的模型以及训练任务。 +创建训练引擎后，如果需要更新密钥或者遇到错误，可以进入应用的设置页面选择“功能”选项选择其中的出厂重启，您的训练程序就会被重置。 ![alt text](/img/engines/autotrain-engine/image.png) -更新完后您就可以启动您的引擎了。 +等待构建和启动完成，你就可以看到训练的界面了。 ![alt text](/img/engines/autotrain-engine/img.png) -### 3.部署训练 +### 3.开始训练 -首先您要选择好您所要训练的主任务类型，以及子任务类型。 +以 bert-base-chinese 为基础，训练一个简单的工作任务分类模型： -![alt text](/img/engines/autotrain-engine/image2.png) +- 首先在页面上选择任务为 `文本分类` ，选择基础模型为 `bert-base-chinese`。 -然后选择您的基础模型，并上传训练数据以及验证数据，最后点击训练。 +![alt text](/img/engines/autotrain-engine/img_2.png) -训练好的模型会保存在您应用的文件目录下，您可以通过 Git 同步进行下载。 +- 按照格式要求，上传准备好的数据 + +![img.png](/img/engines/autotrain-engine/img_3.png) + +- 点击开启训练按钮，即可开始微调任务。 + +您可以通过观察页面下方训练日志了解训练进度。 + +![alt text](/img/engines/autotrain-engine/train-log.png) + +训练结束后，模型会自动上传到您的账号下，可以在个人主页中查看您的模型。 + +![img.png](/img/engines/autotrain-engine/img_4.png) + +您可以在 模型挂件 上进行体验，或者根据你的需要部署到 [模型引擎](/model-engine) 中，对外提供服务。 diff --git a/docs/engines/img.png b/docs/engines/img.png new file mode 100644 index 0000000000000000000000000000000000000000..e1829cc26858e840df5d85934ab5f857be530e7a Binary files /dev/null and b/docs/engines/img.png differ diff --git a/static/img/engines/autotrain-engine/image2.png b/static/img/engines/autotrain-engine/image2.png deleted file mode 100644 index 10ca7a2581b8e453c1a38914d75afa502144bfd9..0000000000000000000000000000000000000000 Binary files a/static/img/engines/autotrain-engine/image2.png and /dev/null differ diff --git a/static/img/engines/autotrain-engine/img_1.png b/static/img/engines/autotrain-engine/img_1.png new file mode 100644 index 0000000000000000000000000000000000000000..5069ed06601c07fd5d567bea61bef3da507eca3e Binary files /dev/null and b/static/img/engines/autotrain-engine/img_1.png differ diff --git a/static/img/engines/autotrain-engine/img_2.png b/static/img/engines/autotrain-engine/img_2.png new file mode 100644 index 0000000000000000000000000000000000000000..ffdf1d10e17d5d0b64c5358343cf8a206fca4d15 Binary files /dev/null and b/static/img/engines/autotrain-engine/img_2.png differ diff --git a/static/img/engines/autotrain-engine/img_3.png b/static/img/engines/autotrain-engine/img_3.png new file mode 100644 index 0000000000000000000000000000000000000000..e1829cc26858e840df5d85934ab5f857be530e7a Binary files /dev/null and b/static/img/engines/autotrain-engine/img_3.png differ diff --git a/static/img/engines/autotrain-engine/img_4.png b/static/img/engines/autotrain-engine/img_4.png new file mode 100644 index 0000000000000000000000000000000000000000..e0cfb0f221c735dc4bf3642a1b2008a0c99658e7 Binary files /dev/null and b/static/img/engines/autotrain-engine/img_4.png differ diff --git a/static/img/engines/autotrain-engine/train-log.png b/static/img/engines/autotrain-engine/train-log.png new file mode 100644 index 0000000000000000000000000000000000000000..ef1e4337fcac4372a33e336532be25a64ffc8235 Binary files /dev/null and b/static/img/engines/autotrain-engine/train-log.png differ