IniziaInizia gratis

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

Visualizza il corso

Istruzioni dell'esercizio

  • Importa Trainer.
  • Definisci il modello ImageClassifier e 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"])
Modifica ed esegui il codice