De Lightning Trainer uitvoeren
Nu je je model hebt gedefinieerd, kun je de Lightning-Trainer configureren om een trainingslus te draaien. In deze oefening leer je hoe je je model, dataloaders en trainingsconfiguraties integreert — deze setup beheersen is essentieel voordat je doorgaat naar geavanceerdere training.
torch en lightning.pytorch, geïmporteerd als pl, zijn alvast voor je geladen. Er is al een model geïnitialiseerd met de klasse die je in de vorige oefening hebt gedefinieerd.
Deze oefening maakt deel uit van de cursus
Schaalbare AI-modellen met PyTorch Lightning
Oefeninstructies
- Configureer de
Trainermet 3 epochs. - Start een trainingsrun met het vooraf gedefinieerde
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)