ITPub博客

首页 > > > Transformers 模型上传和共享 | 五

Transformers 模型上传和共享 | 五

作者:highhand 时间:2020-04-22 06:36:56 0 删除 编辑

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

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

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

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 …复制代码

原文链接:huggingface.co/transformer…

欢迎关注磐创AI博客站: panchuang.net/

OpenCV中文官方文档: woshicver.com/

欢迎关注磐创博客资源汇总站: docs.panchuang.net/

原文作者:;如有侵权,请联系管理员删除。

请登录后发表评论 登录
全部评论

注册时间:2015-11-11

  • 博文量
    1479
  • 访问量
    30521