transformers 模型上传和共享

pytorch • 2020年3月4日 pm12:51 • pytorch, transformers

作者|huggingface
编译|VK
来源|Github

模型上传和共享

从v2.2.2开始，你现在可以使用库中内置的CLI上传和与社区共享你的微调模型。

首先，在以下网址上创建一个帐户。

https://huggingface.co/join

然后：

transformers-cli login
# 使用与huggingface.co上相同的凭据登录

上传你的模型：

transformers-cli upload ./path/to/pretrained_model/

# 上传一个文件夹包含weights/tokenizer/config
# 通过`.save_pretrained()`保存

transformers-cli upload ./config.json [--filename folder/foobar.json]

# 上传单个文件
# (你可以选择覆盖其文件名，该文件名可以嵌套在文件夹中)

然后，可以通过其标识符，用户名和上面的文件夹名称的组合来访问模型：

"username/pretrained_model"

请在model_cards/文件夹下添加README.md到仓库中，其中model_cards/包括：模型描述，训练参数（数据集，预处理，超参数），评估结果，预期用途和限制等。

现在你的模型现在在havingface.co/models上有一个页面🔥

任何人都可以从代码中加载它：

tokenizer = AutoTokenizer.from_pretrained("username/pretrained_model")
model = AutoModel.from_pretrained("username/pretrained_model")

列出你在S3上的所有文件：

transformers-cli s3 ls

你还可以删除不需要的文件：

transformers-cli s3 rm …

原文链接：https://huggingface.co/transformers/model_sharing.html

原创文章，作者：pytorch，如若转载，请注明出处：https://pytorchchina.com/2020/03/04/transformers-%e6%a8%a1%e5%9e%8b%e4%b8%8a%e4%bc%a0%e5%92%8c%e5%85%b1%e4%ba%ab/

赞 (0)
pytorch pytorch
transformers Jupyter Notebooks
« 上一篇 2020年3月4日 pm12:50
transformers 示例教程
下一篇 » 2020年3月4日 pm12:52

相关推荐

QR code