无法从'transformers.modeling_distilbert‘导入名称'DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP’

Question

我正在尝试训练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)

请帮帮我！

Answer 1

我认为这是因为transformer的版本，在我更新transformer===2.7.0之后，它工作了。