Configurar el entrenador
Con los argumentos de entrenamiento configurados, debes pasar al entrenador antes de poner en funcionamiento el modelo afinado.
TrainingArguments y Trainer se han cargado para ti, así como tus anteriores model, tokenizer y training_args.
Nota: en este ejercicio se ha desactivado el entrenamiento real para que se ejecute más rápidamente.
Este ejercicio forma parte del curso
Introduction to LLMs in Python
Instrucciones de ejercicio
- Configura el objeto
Trainer(). - Asigna los argumentos de entrenamiento y el tokenizador previamente definidos.
- Entrena el modelo.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
# Set up the trainer object
trainer = ____(
model=model,
# Assign the training arguments and tokenizer
____,
train_dataset=tokenized_training_data,
eval_dataset=tokenized_test_data,
____
)
# Train the model
____