Hi @ptrblck ,
I have a similar kind of uses case as above to use the pre-trained model of a multi-label classifier.
Single Images has to be classified with two labels.
Used below code to modify the last layer for single-label classifier of 10 classes ,
in_features = resnet.fc.in_features
resnet.fc = nn.Linear(in_features, 10 )
How to modify the last layer for two-label classifier each of 10 classes? I’m a newbie to pytorch
in_features = resnet.fc.in_features
resnet.fc = nn.Linear(in_features, ? )