我正在尝试为问答目的训练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)
请帮助!
答案 0 :(得分:1)
我认为是因为我更新了translator === 2.7.0之后,translator的版本才起作用。