gphilip:

output (which is a list of probabilities) of resnet34


Thanks for the reply.
Sorry, i am not sure if i understood that correctly. Isn’t that  the ‘output (which is a list of probabilities) of resnet34’ will be 20 as the number of my dataset classes are 20, while Imagenet classes are 1000.