Solved by KarthikR in post #2 
              
              
                If requires_grad is set to false, you are freezing the part of the model as no changes happen to its parameters.  In the example below, all layers have the parameters modified during training as requires_grad is set to true. 
import torch, torchvision 
import torch.nn as nn 
from collections import …