Aan de slagGa gratis aan de slag

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.

Deze oefening maakt deel uit van de cursus

Scalable AI Models with PyTorch Lightning

Cursus bekijken

Oefeninstructies

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

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

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)
Code bewerken en uitvoeren