НачатьНачать бесплатно

Обучение и оценка модели

В этом упражнении мы объединим всё, что изучили ранее: обучим нейронную сеть и оценим её на реальном наборе данных рукописных символов эфиопского алфавита MNIST.

ImageClassifier — это заранее определённая модель нейронной сети, реализованная с помощью PyTorch Lightning. Она включает свёрточные слои для извлечения признаков, функции активации для введения нелинейности и полносвязные слои для классификации.

Набор данных Ethiopic MNIST уже импортирован для вас.

Это упражнение является частью курса

Масштабируемые модели ИИ с PyTorch Lightning

Посмотреть курс

Инструкции к упражнению

  • Импортируйте Trainer.
  • Определите модель ImageClassifier и тренер.
  • Запустите обучение модели.
  • Оцените модель на валидационной выборке.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# 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"])
Редактировать и запускать код