Завантаження попередньо натренованої моделі

Ви створюєте застосунок для розмічення зображень із соціальних мереж. Це завдання потребує високої точності та швидкості. Ви використаєте попередньо натреновану модель ResNet18, щоб визначати класи зображень.

Ця вправа є частиною курсу

Глибоке навчання для зображень із PyTorch

Переглянути курс

Інструкції до вправи

Імпортуйте resnet18 і ResNet18_Weights з torchvision.models.
Створіть екземпляр model за допомогою resnet18(), передавши параметр weights як weights.
Переведіть model у режим оцінювання.
Ініціалізуйте вхідні перетворення та присвойте їх змінній transform.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

# Import resnet18 model
from torchvision.models import ____

# Initialize model with default weights
weights = ResNet18_Weights.DEFAULT
model = ____

# Set model to evaluation mode
model.____

# Initialize the transforms
transform = ____

Редагувати та запускати код