Skip to content

Commit 9519e9d

Browse files
Fix warning in fast_neural_style example (#2984)
* Update examples/fast_neural_style/vgg.py * Update examples/fast_neural_style/vgg.py * Update vgg.py --------- Co-authored-by: sweep-ai[bot] <128439645+sweep-ai[bot]@users.noreply.github.com> Co-authored-by: vfdev <vfdev.5@gmail.com>
1 parent 35a2da3 commit 9519e9d

File tree

1 file changed

+2
-1
lines changed
  • examples/fast_neural_style

1 file changed

+2
-1
lines changed

examples/fast_neural_style/vgg.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,13 @@
22

33
import torch
44
from torchvision import models
5+
from torchvision.models.vgg import VGG16_Weights
56

67

78
class Vgg16(torch.nn.Module):
89
def __init__(self, requires_grad=False):
910
super(Vgg16, self).__init__()
10-
vgg_pretrained_features = models.vgg16(pretrained=True).features
11+
vgg_pretrained_features = models.vgg16(weights=VGG16_Weights.IMAGENET1K_V1).features
1112
self.slice1 = torch.nn.Sequential()
1213
self.slice2 = torch.nn.Sequential()
1314
self.slice3 = torch.nn.Sequential()

0 commit comments

Comments
 (0)