невозможно импортировать имя 'DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP' из 'transformers.modeling_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 ответ

Я думаю, это потому, что версия трансформатора, после того как я обновил трансформатор ===2.7.0, работает.

Другие вопросы по тегам