AutoTrain Advanced 是 Hugging Face 开发的无代码机器学习模型训练与部署方案,目标是让训练和部署先进机器学习模型变得更快、更容易。用户无需编写代码,只需上传符合正确格式的数据,即可在几次点击内开始训练模型。
主要能力与用途:该项目提供无代码界面来训练机器学习模型。它强调必须上传格式正确的数据才能创建项目,关于数据格式和定价的帮助可在其文档中查看。文档地址为 https://hf.co/docs/autotrain/。
使用方式:
- 在 Colab 或 Hugging Face Spaces 上运行 UI:可将 AutoTrain 部署到 Hugging Face Spaces,或通过 ngrok 在 Colab 上运行 AutoTrain UI。
- 本地安装:可通过 PIP 安装 AutoTrain-Advanced Python 包,要求 Python >= 3.10。需要安装 git lfs,同时需安装 torch、torchaudio 和 torchvision。推荐在 conda 环境中运行 autotrain,可先创建新的 conda 环境,再启动应用。
- 命令行与配置:如果不喜欢使用 UI,可以使用 AutoTrain Configs 通过命令行训练,或直接使用 AutoTrain CLI。使用配置文件训练时可运行相应命令,示例配置文件位于仓库的 configs 目录,其中包含用于微调 SmolLM2 的示例配置。
收费说明:AutoTrain 是免费的。只有在决定于 Hugging Face Spaces 上运行 AutoTrain 时,才需要为所使用的资源付费。在本地运行时,只需为自身基础设施上使用的资源付费。
维护状态:该项目已不再维护,不会添加新功能,也不会修复错误。官方建议使用 Axolotl、TRL 或 transformers.Trainer 替代。
限制与说明:资料中未提供具体支持的任务列表、支持的模型类型、界面截图或详细功能说明;价格、性能和效果等信息暂未提供。