Eseguire il Lightning Trainer
Ora che hai definito il tuo modello, sei pronto per configurare il Trainer di Lightning per eseguire un ciclo di training. Questo esercizio ti aiuta a capire come integrare modello, data loader e configurazioni di training: padroneggiare questo setup è fondamentale prima di passare a tecniche più avanzate.
torch e lightning.pytorch, importato come pl, sono già caricati per te. Un model è già stato inizializzato usando la classe che hai definito nell'esercizio precedente.
Questo esercizio fa parte del corso
Modelli di AI scalabili con PyTorch Lightning
Istruzioni dell'esercizio
- Configura il
Trainercon 3 epoche. - Avvia un training run con il
modelpredefinito.
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)