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.

Questo esercizio fa parte del corso

Deep Learning for Images with PyTorch

Visualizza il corso

Istruzioni dell'esercizio

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.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

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

Modifica ed esegui il codice