Проблема с загрузкой базовой модели Roberta

#python #pytorch #roberta-language-model

#питон #пыторч #роберта-языковая модель

Вопрос:

Я пытаюсь использовать базовую модель Roberta, используя AutoTokenizer.from_pretrained('roberta-base') , но я получаю следующую ошибку:

 RuntimeError: Failed to import transformers.modeling_tf_utils because of the following error (look up to see its traceback):
No module named 'tensorflow.python.keras.engine.keras_tensor'
 

Я попытался установить tensorflow, но все та же ошибка. Есть идеи, что происходит?

Ответ №1:

Вам необходимо установить с помощью GPU, попробуйте это:

 pip install --ignore-installed --upgrade tensorflow-gpu
 

Смотрите Здесь для получения более подробной информации.

Комментарии:

1. отлично работает, спасибо

2. отлично работает, спасибо