Hello,
I am curious about the layer naming (key values of state_dict) of the vgg16 pretrained model from torchvision.models module, e.g.: ‘features.0.weight’, ‘features.0.bias’, ‘features.2.weight’, ‘features.2.bias’, etc. The number increases by 2 with each new convolutional or fc layer, and increases by 3 each time encountering a max pooling layer.
Do you have an idea of the underlying logic?
Thank you very much!
Guangye