Pytorch Resnet50 Model