@ptrblck thanks a lot for the reply. I first downloaded tiny-imagenet dataset which has 200 classes and each with 500 images from imagenet webpage then in code I get the resnet101 model from torchvision.models and perform inference on the train folder of tiny-imagenet. which provides only 18% accuracy as I mentioned earlier.