Why pre-trained models accelerate development in PyTorch - Test Your Understanding

Practice - 5 Tasks

1fill in blank

easy

Complete the code to load a pre-trained model in PyTorch.

PyTorch

import torch
import torchvision.models as models

model = models.resnet18(pretrained=[1])

Drag options to blanks, or click blank then click option'

ATrue

BFalse

CNone

D0

Attempts:

3 left

💡 Hint

Common Mistakes

Using False will create a model with random weights.

Setting pretrained=True loads the model with weights trained on a large dataset, speeding up development.

2fill in blank

medium

Complete the code to freeze all layers of the pre-trained model.

PyTorch

for param in model.[1]():
    param.requires_grad = False

Drag options to blanks, or click blank then click option'

Alayers

Bchildren

Cmodules

Dparameters

Attempts:

3 left

💡 Hint

Common Mistakes

Using children() returns submodules, not parameters.

The parameters() method returns all parameters of the model to freeze them.

3fill in blank

hard

Fix the error in replacing the final layer for transfer learning.

PyTorch

import torch.nn as nn
num_classes = 10
model.fc = nn.[1](num_classes)

Drag options to blanks, or click blank then click option'

AConv2d

BLinear

CReLU

DDropout

Attempts:

3 left

💡 Hint

Common Mistakes

Using Conv2d or activation layers instead of Linear.

The final fully connected layer should be replaced with nn.Linear to match the number of classes.

4fill in blank

hard

Fill both blanks to set the model to evaluation mode and disable gradient calculation.

PyTorch

model.[1]()
with torch.[2]():
    outputs = model(inputs)

Drag options to blanks, or click blank then click option'

Aeval

Bno_grad

Ctrain

Denable_grad

Attempts:

3 left

💡 Hint

Common Mistakes

Using train() instead of eval(), or enable_grad instead of no_grad.

Use eval() to set evaluation mode and no_grad() to disable gradients during inference.

5fill in blank

hard

Fill all three blanks to create a dictionary of layer names and their requires_grad status.

PyTorch

grad_status = {name: param.[1] for name, param in model.[2]() if name.[3]('fc') == False}

Drag options to blanks, or click blank then click option'

Arequires_grad

Bnamed_parameters

Cendswith

Dstartswith

Attempts:

3 left

💡 Hint

Common Mistakes

Using parameters() instead of named_parameters(), or endswith instead of startswith.

This code collects the gradient status of all layers except those starting with 'fc' (final layer).

Practice

1. Why do pre-trained models help speed up AI development in PyTorch?

easy

A. They always produce perfect results without any training.

B. They start with knowledge learned from other data, reducing training time.

C. They require more data to train from scratch.

D. They avoid the need for any coding or model building.

Solution

1. Step 1: Understand pre-trained model concept

   Pre-trained models have already learned patterns from large datasets, so they don't start from zero.

2. Step 2: Relate to training time

   Because they start with learned features, training on new tasks is faster and needs less data.

3. Final Answer:

   They start with knowledge learned from other data, reducing training time. -> Option B

4. Quick Check:

   Pre-trained models speed development by reusing learned knowledge [OK]

Hint: Pre-trained means already learned, so less training needed [OK]

Common Mistakes:

• Thinking pre-trained models need more data
• Believing pre-trained models don't require any training
• Assuming pre-trained models are perfect without fine-tuning

2. Which PyTorch code snippet correctly loads a pre-trained ResNet model?

easy

A. model = torchvision.models.resnet50(weights='IMAGENET1K_V1')

B. model = torchvision.models.resnet50(pretrained=False)

C. model = torchvision.models.resnet50(pretrained=false)

D. model = torchvision.models.resnet50(load_pretrained=True)

Solution

1. Step 1: Check PyTorch's current API for loading pre-trained models

   Recent PyTorch versions use the 'weights' parameter to specify pre-trained weights, e.g., weights='IMAGENET1K_V1'.

2. Step 2: Identify correct syntax

   model = torchvision.models.resnet50(weights='IMAGENET1K_V1') uses 'weights="IMAGENET1K_V1"', which is the correct way to load pre-trained weights in PyTorch 1.12+.

3. Final Answer:

   model = torchvision.models.resnet50(weights='IMAGENET1K_V1') -> Option A

4. Quick Check:

   Use weights='IMAGENET1K_V1' to load pre-trained models [OK]

Hint: Use weights='IMAGENET1K_V1' for pre-trained models in PyTorch 1.12+ [OK]

Common Mistakes:

• Using deprecated pretrained=True parameter
• Using nonexistent load_pretrained argument
• Setting pretrained=False which loads untrained model

3. What will be the output shape of the final layer when fine-tuning a pre-trained ResNet50 model for 10 classes in PyTorch?

medium

A. [batch_size, 10]

B. [batch_size, 512]

C. [10, batch_size]

D. [batch_size, 1000]

Solution

1. Step 1: Understand ResNet50 default output

   By default, ResNet50 outputs 1000 classes for ImageNet classification.

2. Step 2: Fine-tuning changes final layer output size

   When fine-tuning for 10 classes, the final fully connected layer is replaced to output 10 values per input.

3. Final Answer:

   [batch_size, 10] -> Option A

4. Quick Check:

   Fine-tuned model outputs match new class count [OK]

Hint: Final layer output matches number of classes [OK]

Common Mistakes:

• Assuming output stays 1000 classes after fine-tuning
• Confusing batch size and class dimension order
• Using feature size (512) as output shape

4. You tried to fine-tune a pre-trained model but get a shape mismatch error on the last layer. What is the likely cause?

medium

A. The model was not loaded with pre-trained weights.

B. The optimizer learning rate is too high.

C. The input images are not normalized correctly.

D. The final layer's output size does not match the new task's number of classes.

Solution

1. Step 1: Identify cause of shape mismatch error

   Shape mismatch usually happens when the model's last layer output size differs from the target labels size.

2. Step 2: Relate to fine-tuning process

   When fine-tuning, you must replace the last layer to match the new number of classes; otherwise, shapes won't align.

3. Final Answer:

   The final layer's output size does not match the new task's number of classes. -> Option D

4. Quick Check:

   Shape mismatch means output layer size differs from labels [OK]

Hint: Check last layer output size matches target classes [OK]

Common Mistakes:

• Blaming optimizer or input normalization for shape errors
• Forgetting to replace the final layer for new tasks
• Assuming pre-trained weights cause shape mismatch

5. You have a small dataset and limited GPU power. How does using a pre-trained model in PyTorch help you build an accurate classifier faster?

hard

A. It automatically generates more data to train on.

B. It trains the entire model from scratch faster than a new model.

C. It allows you to fine-tune only the last layers, reducing training time and data needs.

D. It removes the need for validation and testing.

Solution

1. Step 1: Understand constraints of small data and limited GPU

   Training a full model from scratch requires lots of data and computing power, which are limited here.

2. Step 2: Explain benefit of fine-tuning pre-trained models

   Pre-trained models have learned features already, so you can train only the last layers, saving time and data.

3. Step 3: Why other options are incorrect

   It trains the entire model from scratch faster than a new model. is wrong because training from scratch is slower. It automatically generates more data to train on. is false; pre-trained models don't generate data. It removes the need for validation and testing. is incorrect; validation/testing are always needed.

4. Final Answer:

   It allows you to fine-tune only the last layers, reducing training time and data needs. -> Option C

5. Quick Check:

   Fine-tuning last layers saves time and data [OK]

Hint: Fine-tune last layers to save time and data [OK]

Common Mistakes:

• Thinking pre-trained models generate more data
• Believing full training is faster than fine-tuning
• Skipping validation/testing phases