【发布时间】:2021-01-09 05:51:38
【问题描述】:
尝试从 simpletransformers 导入分类模型时出现以下错误。
ImportError Traceback (most recent call last)
<ipython-input-1-29f08e6c2d87> in <module>()
----> 1 from simpletransformers.classification import ClassificationModel, ClassificationArgs
3 frames
/usr/local/lib/python3.6/dist-packages/simpletransformers/classification/transformer_models/roberta_model.py in <module>()
2 import torch.nn as nn
3 from torch.nn import CrossEntropyLoss, MSELoss
----> 4 from transformers.modeling_roberta import (
5 ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST,
6 BertPreTrainedModel,
ImportError: cannot import name 'BertPreTrainedModel'
---------------------------------------------------------------------------
【问题讨论】:
-
from pytorch_pretrained_bert.modeling import BertPreTrainedModel
标签: keras pytorch tensorflow2.0 bert-language-model huggingface-transformers