Oh! This minimal code is working for me.
import torch
import torchvision.models.vgg as models

input = torch.rand(1, 3, 5, 5)
vgg16 = models.vgg16(pretrained=True)
output = vgg16.features[:3](input)
print(output)

I’m using PyTorch 0.4.0 to run this.