【问题标题】:Getting an error(cannot import name 'BertPreTrainedModel') while importing classification model from simpletransformers从 simpletransformers 导入分类模型时出现错误(无法导入名称“BertPreTrainedModel”)
【发布时间】: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


【解决方案1】:

this github 问题中,问题是simpletransformers 的旧版本。要获取最新版本,请执行pip install --upgrade simpletransformers。甚至可能对 transformers 包也这样做。

【讨论】:

  • 似乎最新版本的转换器不能很好地与 simpletransformers 一起使用。因此不得不将它们都降级到以前的版本。 !pip 安装变压器==3.1.0 ; !pip install simpletransformers==0.48.1。降级后它工作正常。
猜你喜欢
  • 1970-01-01
  • 1970-01-01
  • 2016-12-03
  • 2021-03-11
  • 2019-03-10
  • 2013-02-22
  • 1970-01-01
相关资源
最近更新 更多