Ejecutar Lightning Trainer
Ahora que ya tienes definido el modelo, estás listo para configurar Lightning Trainer
para ejecutar un bucle de entrenamiento. Este ejercicio te ayuda a comprender cómo integrar tu modelo, los cargadores de datos y las configuraciones de entrenamiento. Es fundamental dominar esta configuración antes de pasar a un entrenamiento más avanzado.
Los archivos « torch
» y « lightning.pytorch
», importados como « pl
», se han precargado para ti. Ya se ha inicializado un objeto de clase ` model
`, utilizando la clase que definiste en el ejercicio anterior.
Este ejercicio forma parte del curso
Modelos de IA escalables con PyTorch Lightning
Instrucciones del ejercicio
- Configura el algoritmo de aprendizaje por refuerzo (
Trainer
) con 3 épocas. - Inicia una sesión de entrenamiento con el programa predefinido «
model
» (Entrenamiento de carrera de 10 km).
Ejercicio interactivo práctico
Prueba este ejercicio completando el código de muestra.
from lightning.pytorch import Trainer
# Configure the Trainer with max epochs
trainer = ____(max_epochs=____, devices=1)
# Start training with the predefined model
trainer.____(____, train_dataloader, val_dataloader)