The pretrained weights were added in this PR last year so you might need to update torchvision
.
This code works fine in the current nightly release:
model = models.shufflenet_v2_x2_0(pretrained=True)
1 Like