From af3ef43f011c6c1d811d87c013f47de110bb7f0b Mon Sep 17 00:00:00 2001 From: alperenunlu Date: Sun, 12 Nov 2023 17:34:41 +0300 Subject: [PATCH] Fix weight initialization on neural_style_tutorial --- advanced_source/neural_style_tutorial.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/advanced_source/neural_style_tutorial.py b/advanced_source/neural_style_tutorial.py index ee4dab7e7ec..a293026cb5b 100644 --- a/advanced_source/neural_style_tutorial.py +++ b/advanced_source/neural_style_tutorial.py @@ -56,7 +56,7 @@ import matplotlib.pyplot as plt import torchvision.transforms as transforms -import torchvision.models as models +from torchvision.models import vgg19, VGG19_Weights import copy @@ -262,7 +262,7 @@ def forward(self, input): # network to evaluation mode using ``.eval()``. # -cnn = models.vgg19(pretrained=True).features.eval() +cnn = vgg19(weights=VGG19_Weights.DEFAULT).features.eval()