PyTorch中如何在本地加载Roberta模型:Transformers实战指南

本地加载roberta-base模型文件,roberta-large同理,

只不过hidden_size从768变为1024,

在该网站下载模型文件:
roberta-base at main (huggingface.co)

所需的有 config.json,  merges.txt,  pytorch_model.bin(下载后重命名),  vocab.json

路径组织结构:

└─model

│  merges.txt

│  vocab.json

└─roberta-base

config.json

pytorch_model.bin

此处transformers版本是2.11.0,

不同版本的transformers里import的用法可能不同,所以强调了版本,

可针对自己的版本去看官方文档

from transformers import RobertaTokenizer

vocab_file = 'model/vocab.json'

merges_file = 'model/merges.txt'

tokenizer = RobertaTokenizer(vocab_file, merges_file)

RobertaTokenizer 和 BertTokenizer 处理数据时不一样,

Roberta模型没有token_type_ids,也就是segment_ids

还有 Bert 的数据格式是

  • pair of sequences:
    [CLS] A [SEP] B [SEP]

Roberta 的则是

  • pair of sequences:
    <s> A </s></s> B </s>

参见:
RobertaTokenizer中的build_inputs_with_special_tokens


BertTokenizer中的build_inputs_with_special_tokens

from transformers.modeling_roberta import RobertaModel

 bert = RobertaModel.from_pretrained('model/roberta-base/')

相关推荐

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-21 23:34:02       52 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-21 23:34:02       54 阅读
  3. 在Django里面运行非项目文件

    2024-07-21 23:34:02       45 阅读
  4. Python语言-面向对象

    2024-07-21 23:34:02       55 阅读

热门阅读

  1. Shiro-550反序列化漏洞

    2024-07-21 23:34:02       15 阅读
  2. Kotlin单例、数据类、静态

    2024-07-21 23:34:02       17 阅读
  3. CSP-J模拟赛day1

    2024-07-21 23:34:02       19 阅读
  4. Linux下双网卡NAT组网

    2024-07-21 23:34:02       18 阅读
  5. Node的API基础

    2024-07-21 23:34:02       17 阅读
  6. C2W3.LAB.N-grams+Language Model+OOV

    2024-07-21 23:34:02       17 阅读
  7. 力扣题解(一和零)

    2024-07-21 23:34:02       20 阅读
  8. urllib&requests

    2024-07-21 23:34:02       15 阅读