Configurando un Trainer para computer vision

Para quienes deseen guardar su modelo en un repo personal deben eliminar el argumento push_to_organization y cambiar el repositorio destino a uno personal, así output_dir="https://huggingface.co/NOMBRE_USUARIO/vit_model"

Es decir:

training_args = TrainingArguments(
    output_dir="https://huggingface.co/santiagoahl/vit_model",
    evaluation_strategy='steps',
    num_train_epochs=4,
    learning_rate=2e-4,
    remove_unused_columns=False,
    push_to_hub=True,
    load_best_model_at_end=True
)

Si a alguno se le presenta error al momento de instanciar training_args

ImportError: Using the `Trainer` with `PyTorch` requires `accelerate>=0.20.1`: Please run `pip install transformers[torch]` or `pip install accelerate -U`

Yo lo solucioné creando un entorno virtual en colab e instalando las versiones compatibles de transformer y accelerate

!pip install virtualenv

# Crear un nuevo entorno virtual
!python3 -m virtualenv myenv

# Activar el entorno virtual
!source myenv/bin/activate

# Instalar las bibliotecas necesarias
!pip install transformers==4.30.2
!pip install accelerate==0.20.1

Les comparto el Hub de Platzi en Hugging Face 🤗
https://huggingface.co/platzi

Hay que estar suscrito a la organización de Platzi para poder hacer envio del modelo.

Si alguno está teniendo problemas con la versión de transformers, revisen este issue: https://github.com/huggingface/transformers/issues/22816

Ejecutar todo con una versión más baja de transformers fue la solución.

el dialogo de la sugerencia de documentación del colab es un poco molesto, sería mejor que el profesor lo esconda ya que a veces cuesta leer lo que escribe

Gracias