Training ed evaluation
In questo esercizio metteremo insieme tutto ciò che abbiamo praticato finora, addestrando e valutando una rete neurale sul dataset reale di caratteri etiopi MNIST scritti a mano.
ImageClassifier è un modello di rete neurale predefinito implementato con PyTorch Lightning. È composto da livelli convoluzionali per l’estrazione delle caratteristiche, funzioni di attivazione per introdurre non linearità e livelli completamente connessi per la classificazione.
Il dataset Ethiopic MNIST è già stato importato per te.
Questo esercizio fa parte del corso
Modelli di AI scalabili con PyTorch Lightning
Istruzioni dell'esercizio
- Importa
Trainer. - Definisci il modello
ImageClassifiere il trainer. - Allena il modello.
- Valuta il modello sul set di validazione.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Import the Trainer
from lightning.pytorch import ____
# Define ImageClassifier model & trainer and set epoch parameter
model = ____()
trainer = ____(max_epochs=5)
# Train the model
trainer.fit(____, train_loader, val_loader)
# Evaluate the model
val_results = trainer.____(____, val_loader)
print("Validation Accuracy:", val_results[0]["val_acc"])