@SelvamArul
I want to extract feature vectors from the last hidden layer (before the softmax layer).
I don’t want replace the fc layer like resnet.fc = nn.Linear(resnet.fc.in_features, embed_size). I don’t want train the resnet fc layer,  just want to extract the feature vectors from it.
Thanks