无法从“ transformers.modeling_distilbert”导入名称“ DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP”

时间:2020-06-03 12:33:14

标签: python neural-network bert-language-model question-answering distilbert

我正在尝试为问答目的训练Distil BERT模型。 我已经安装了简单的转换器以及所有东西,但是当我尝试运行以下命令时:

model = QuestionAnsweringModel('distilbert', 
                               MODEL_PATH, 
                              args={'reprocess_input_data': True,
                                    'overwrite_output_dir': True,
                                    'learning_rate': 5e-5,
                                     'num_train_epochs': 2,
                                     'max_seq_length': 192,
                                   'doc_stride': 64,
                                   'fp16': False
                                  }, use_cuda=use_cuda)

我遇到错误->

cannot import name 'DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP' 
from 'transformers.modeling_distilbert' 
(/opt/conda/lib/python3.7/site-packages/transformers/modeling_distilbert.py)

请帮助!

1 个答案:

答案 0 :(得分:1)

我认为是因为我更新了translator === 2.7.0之后,translator的版本才起作用。