es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

No se puede importar el nombre ‘TrainingArguments’ de ‘transformers’.

29 May, 2023 Programación

Estoy intentando ajustar un modelo BERT preentrenado de HuggingFace. Estoy importando lo siguiente:

from transformers import (AutoTokenizer, AutoConfig, 
                              AutoModelForSequenceClassification, TrainingArguments, Trainer)

y obtengo el siguiente error:

no se puede importar el nombre 'TrainingArguments' de 'transformers'

Tampoco puedo importar ‘Trainer’.

Actualmente tengo instalados tensorflow 2.2.0, pytorch 1.7.1, y transformers 2.1.1.

Tags:  , ,

Answer

  1. Avatar for davy.ai
    davy.ai

    Parece que el módulo TrainingArguments no está disponible en transformers 2.1.1. Este módulo fue añadido en la versión 2.3.0 de transformers. Por lo tanto, debe actualizar su biblioteca de transformers a la versión 2.3.0 o superior.

    Para hacer esto, puede ejecutar el siguiente comando en su terminal:

    pip install --upgrade transformers

    Después de actualizar, debería poder importar TrainingArguments y Trainer con el siguiente código:

    from transformers import (AutoTokenizer, AutoConfig,
                                  AutoModelForSequenceClassification, TrainingArguments, Trainer)

    Asegúrese también de verificar si se necesitan actualizar otras dependencias para trabajar con la versión más reciente de transformers.

Comments are closed.