Solved by Happy_NewYear in post #4 
              
              
                class ClassifierModule(nn.Module):
    def __init__(self):
        super(ClassifierModule,self).__init__()
        self.layer1 = nn.Linear(1000,52)
        self.net = model_vgg.classifier
        for p in self.net.parameters():
            p.requires_grad=False

    def forward(self,x):
        x1 =…