Formation et évaluation
Dans cet exercice, nous allons rassembler tout ce que nous avons appris jusqu'à présent en entraînant et en évaluant un réseau neuronal sur un ensemble de données réelles de caractères MNIST éthiopiens manuscrits.
ImageClassifier
est un modèle de réseau neuronal prédéfini implémenté à l'aide de PyTorch Lightning. Il se compose de couches convolutives pour l'extraction des caractéristiques, de fonctions d'activation pour introduire la non-linéarité et de couches entièrement connectées pour la classification.
L'ensemble de données MNIST en éthiopien a été pré-importé pour vous.
Cet exercice fait partie du cours
Modèles d'IA évolutifs avec PyTorch Lightning
Instructions
- Veuillez importer l'
Trainer
. - Définissez le modèle d'
ImageClassifier
et le formateur. - Entraînez le modèle.
- Évaluez le modèle sur l'ensemble de validation.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# 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"])