AutoTrain 文档

Hugging Face Spaces 上 AutoTrain 快速入门指南

您正在查看 main 版本,该版本需要从源代码安装。如果您想要常规的 pip 安装,请查看最新的稳定版本 (v0.8.24)。
Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始使用

Hugging Face Spaces 上 AutoTrain 快速入门指南

在 Hugging Face Spaces 上使用 AutoTrain 是模型训练精简体验的首选。此平台针对易用性进行了优化,具有预安装的依赖项和托管的硬件资源。Hugging Face Spaces 上的 AutoTrain 可以供无代码用户和开发人员使用,使其适用于各种专业水平。

创建新的 AutoTrain Space

AutoTrain 入门非常简单。以下是如何创建新的 space:

  1. 访问 AutoTrain 页面:要使用 AutoTrain Docker 镜像创建新的 space,您只需访问 AutoTrain 首页 并点击“创建新项目”。

  2. 登录或查看设置屏幕:如果未登录,系统将提示您登录。然后,您将看到类似于这样的屏幕

autotrain-duplicate-space

  1. 设置您的 Space:
  • 选择 Space 名称:将您的 space 命名为与您的项目相关的内容。

  • 分配硬件资源:根据您的项目需求选择必要的计算资源。

  • 复制 Space:点击“复制 Space”以使用 Docker 镜像启动您的 AutoTrain space。

  1. 配置选项:
  • PAUSE_ON_FAILURE:如果您希望 space 在训练失败时不暂停,请将其设置为 0,这对于运行持续实验很有用。如果您希望在同一 space 中持续执行多个实验,也可以使用此选项。
  1. 启动并训练:
  • 完成后,几秒钟内,AutoTrain Space 将启动并运行,您将看到以下屏幕

autotrain-space

  • 在这里,您可以选择任务、上传数据集、选择模型、调整超参数(如果需要),并直接在 space 内开始训练过程。

  • space 将管理其自身的活动,在训练后关闭,除非根据 PAUSE_ON_FAILURE 设置另行配置。

  1. 监控进度:
  • 所有训练日志和进度都可以通过 TensorBoard 监控,可在 Hugging Face Hub 上的 username/project_name 下访问。

  • 训练成功结束后,您将在同一存储库中找到模型文件。

  1. 导航 UI:
  • 如果您需要帮助理解任何 UI 元素,请点击小的 (i) 信息图标以获取详细描述。

如果您对 UI 元素感到困惑,请点击小的 (i) 信息图标以获取有关 UI 元素的更多信息。

有关数据格式和详细参数信息,请参阅“数据格式和参数”部分,我们在其中提供了示例数据集以及 AutoTrain 支持的每个任务的详细参数信息。

确保您的 AutoTrain 是最新的

我们不断向 AutoTrain Advanced 添加新功能和任务。为了受益于最新的功能、任务和错误修复,请定期更新您的 AutoTrain space

  • 恢复出厂设置:导航到您的 space 的设置页面,然后点击“恢复出厂设置”以升级到最新版本的 AutoTrain Advanced。

autotrain-space-template

  • 注意:仅仅“重启” space 不会更新它;要完全更新,必须进行恢复出厂设置。

有关数据格式和特定参数的更多详细信息,请参阅“数据格式和参数”部分,我们在其中提供了示例数据集以及 AutoTrain 支持的每个任务的广泛参数信息。

通过这些步骤,您可以轻松地在 Hugging Face Spaces 上启动和管理您的 AutoTrain 项目,利用该平台强大的功能来满足您的机器学习和 AI 需求。

< > 在 GitHub 上更新