Zacznij terazZacznij za darmo

Trenowanie i ewaluacja

W tym ćwiczeniu połączysz wszystko, czego się do tej pory uczyłeś – wytrenuj i oceń sieć neuronową na rzeczywistym zbiorze danych odręcznie pisanych znaków etiopskiego alfabetu MNIST.

ImageClassifier to gotowy model sieci neuronowej zaimplementowany przy użyciu PyTorch Lightning. Składa się z warstw splotowych do ekstrakcji cech, funkcji aktywacji wprowadzających nieliniowość oraz w pełni połączonych warstw do klasyfikacji.

Zbiór danych Ethiopic MNIST został już zaimportowany.

To ćwiczenie jest częścią kursu

Skalowalne modele AI z PyTorch Lightning

Zobacz kurs

Instrukcje do ćwiczenia

  • Zaimportuj Trainer.
  • Zdefiniuj model ImageClassifier i obiekt trenera.
  • Wytrenuj model.
  • Oceń model na zbiorze walidacyjnym.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# 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"])
Edytuj i uruchom kod