There is currently an extra layer at the end of the architecture. This does not figure on the `master `branch or here https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py