Loading a pre-trained model

You are building an application to label images from the social media. This task requires high accuracy and speed. You are going to use a pre-trained ResNet18 model to infer image classes.

This exercise is part of the course

Deep Learning for Images with PyTorch

View Course

Exercise instructions

Import resnet18 and ResNet18_Weights from torchvision.models.
Instantiate the model using resnet18(), setting the weights parameter to weights.
Set model to the evaluation mode.
Initialize the input transforms and assign them to transform.

Hands-on interactive exercise

Have a go at this exercise by completing this sample code.

# 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 = ____

Edit and Run Code