Transformers 模型上传和共享 | 五
本文是全系列中第6 / 13篇:Transformers
- Transformers 词汇表 | 二
- Transformers 从pytorch-pretrained-bert迁移 | 十
- Transformers 库常见的用例 | 三
- Transformers 多语言模型 | 十一
- Transformers 中使用 TorchScript | 四
- Transformers 模型上传和共享 | 五
- Transformers 示例 | 六
- Transformers 简介(上)
- Transformers 加载预训练模型 | 七
- Transformers 简介(下)
- Transformers 保存并加载模型 | 八
- Transformers 快速入门 | 一
- Transformers 转换Tensorflow的Checkpoints | 九
作者|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
原创文章,作者:磐石,如若转载,请注明出处:https://panchuang.net/2020/03/28/transformers-%e6%a8%a1%e5%9e%8b%e4%b8%8a%e4%bc%a0%e5%92%8c%e5%85%b1%e4%ba%ab-%e4%ba%94/