Hello!
I am using Squeezenet 1_1 Model pretrained model to train a custom dataset with 4 classes. I used the following commands to instantiate the model
MODEL = models.squeezenet1_1(pretrained=True)
MODEL.classifier[1] = nn.Conv2d(512, self.num_classes, kernel_size=(1, 1), stride=(1, 1))
Then, I write my training script, which is quite boilerplate. I do not freeze any layers.
So, is only the Conv2d layer getting trained with new weights? Do the MaxPool2d layers in Squeezenet have the old weights? Is there a command which we can use that gives the weights of the entire net before and after the training process?