from_pretrained怎么加载训练好的模型
时间: 2024-01-07 20:02:31 浏览: 338
PyPI 官网下载 | pytorch_pretrained_bert-0.3.0.tar.gz
5星 · 资源好评率100%
使用`from_pretrained`方法加载训练好的模型,可以从本地文件或者在线资源中加载预训练的模型。以下是一个简单的示例代码:
```python
from transformers import AutoTokenizer, AutoModel
# 加载 BERT 模型和 tokenizer
tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased')
model = AutoModel.from_pretrained('bert-base-uncased')
# 使用模型和 tokenizer 进行推理
inputs = tokenizer("Hello, world!", return_tensors="pt")
outputs = model(**inputs)
```
在上面的示例中,我们使用`from_pretrained`方法加载了一个预训练的 BERT 模型和相应的 tokenizer,并使用它们进行推理。`from_pretrained`方法可以从 Hugging Face 提供的模型仓库中自动下载预训练模型,也可以从本地文件系统中加载预训练模型。需要注意的是,模型和 tokenizer 的名称必须和从模型仓库下载的名称保持一致。
阅读全文