ComenzarEmpieza gratis

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

Ver curso

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)
Editar y ejecutar código