IniziaInizia gratis

Running the Lightning Trainer

Now that you have your model defined, you are ready to configure the Lightning Trainer to run a training loop. This exercise helps you understand how to integrate your model, data loaders, and training configurations - mastering this setup is essential before moving on to more advanced training.

The torch and lightning.pytorch, imported as pl, have been preloaded for you. A model has already been initialized, using the class you defined in the previous exercise.

Questo esercizio fa parte del corso

Scalable AI Models with PyTorch Lightning

Visualizza il corso

Istruzioni dell'esercizio

  • Configure the Trainer with 3 epochs.
  • Start a training run with the predefined model.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

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)
Modifica ed esegui il codice