Thanks for the reply. When I print, I get this:
odict_keys([‘conv1.0.weight’, ‘conv1.0.bias’, ‘conv2.0.weight’, ‘conv2.0.bias’, ‘conv3.0.weight’, ‘conv3.0.bias’, ‘conv4.0.weight’, ‘conv4.0.bias’, ‘conv5.0.weight’, ‘conv5.0.bias’, ‘conv51.weight’, ‘conv51.bias’, ‘conv52.weight’, ‘conv52.bias’, ‘conv53.weight’, ‘conv53.bias’, ‘fc.0.weight’, ‘fc.0.bias’, ‘fc.2.weight’, ‘fc.2.bias’, ‘conv6.weight’, ‘conv6.bias’, ‘conv7.weight’, ‘conv7.bias’, ‘dconv1.2.weight’, ‘dconv1.2.bias’, ‘dconv2.2.weight’, ‘dconv2.2.bias’, ‘dconv3.2.weight’, ‘dconv3.2.bias’, ‘dconv4.2.weight’, ‘dconv4.2.bias’, ‘conv8.2.weight’, ‘conv8.2.bias’, ‘conv9.2.weight’, ‘conv9.2.bias’])