diff --git a/examples/fast_neural_style/vgg.py b/examples/fast_neural_style/vgg.py index bff7e569329..b8dadeedc21 100644 --- a/examples/fast_neural_style/vgg.py +++ b/examples/fast_neural_style/vgg.py @@ -4,6 +4,7 @@ from torchvision import models from torchvision.models.vgg16 import VGG16_Weights + class Vgg16(torch.nn.Module): def __init__(self, requires_grad=False): super(Vgg16, self).__init__()