Mulai sekarangMulai gratis

Melatih dan mengevaluasi

Pada latihan ini, Anda akan menggabungkan semua yang telah dipraktikkan sejauh ini dengan melatih dan mengevaluasi jaringan saraf pada himpunan data dunia nyata berisi karakter tulisan tangan Ethiopic MNIST.

ImageClassifier adalah model jaringan saraf yang sudah didefinisikan dan diimplementasikan menggunakan PyTorch Lightning. Model ini terdiri atas lapisan konvolusional untuk ekstraksi fitur, fungsi aktivasi untuk memperkenalkan nonlinieritas, serta lapisan terhubung penuh untuk klasifikasi.

Himpunan data Ethiopic MNIST telah diimpor untuk Anda.

Latihan ini merupakan bagian dari kursus

Model AI yang Dapat Diskalakan dengan PyTorch Lightning

Lihat Kursus

Instruksi latihan

  • Impor Trainer.
  • Definisikan model ImageClassifier dan trainer.
  • Latih model.
  • Evaluasi model pada himpunan validasi.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

# 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"])
Edit dan Jalankan Kode