🤗 AutoTrain 高级版
AutoTrain 高级版:更快速、更便捷的训练和部署最先进的机器学习模型。AutoTrain 高级版是一种无需编码的解决方案,只需点击几下即可训练机器学习模型。请注意,您必须上传正确格式的数据才能创建项目。有关正确数据格式和定价的帮助,请查看文档。
注意:AutoTrain 是免费的!只有在您决定在 Hugging Face Spaces 上运行 AutoTrain 时才需为使用的资源付费。在本地运行时,您只需为自己基础设施上使用的资源付费。
在 Colab 或 Hugging Face Spaces 上运行
本地安装
您可以通过 PIP 安装 AutoTrain-Advanced Python 包。请注意,您需要 Python >= 3.10 以确保 AutoTrain 高级版正常工作。
pip install autotrain-advanced
请确保您已安装 git lfs。安装说明请查看:https://github.com/git-lfs/git-lfs/wiki/Installation
您还需要安装 torch、torchaudio 和 torchvision。
运行 autotrain 的最佳方式是在 conda 环境中。您可以使用以下命令创建新的 conda 环境:
conda create -n autotrain python=3.10
conda activate autotrain
pip install autotrain-advanced
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc
完成后,您可以使用以下命令启动应用程序:
autotrain app --port 8080 --host 127.0.0.1
如果您不喜欢使用 UI,可以使用 AutoTrain 配置文件通过命令行进行训练,或者直接使用 AutoTrain CLI。
要使用配置文件进行训练,可以使用以下命令:
autotrain --config <path_to_config_file>
您可以在本存储库的 configs
目录中找到示例配置文件。