Začněte nyníZačněte zdarma

Trénink a vyhodnocení

V tomto cvičení propojíme vše, co jsme si dosud procvičili – natrénujeme a vyhodnotíme neuronovou síť na reálném datasetu ručně psaných etiopských znaků MNIST.

ImageClassifier je předdefinovaný model neuronové sítě implementovaný pomocí PyTorch Lightning. Skládá se z konvolučních vrstev pro extrakci příznaků, aktivačních funkcí pro zavedení nelinearity a plně propojených vrstev pro klasifikaci.

Dataset Ethiopic MNIST je již pro tebe naimportovaný.

Toto cvičení je součástí kurzu

Škálovatelné modely AI s PyTorch Lightning

Zobrazit kurz

Pokyny k cvičení

  • Importuj Trainer.
  • Definuj model ImageClassifier a trainer.
  • Natrénuj model.
  • Vyhodnoť model na validační sadě.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# 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"])
Upravit a spustit kód