I trained the model by using python 2.7 version of pytorch using “pip”. During testing I used a python 3.6 version of pytorch which was installed using conda. I guess the torchvision is different in the pip and conda installation procedure.