replaced deprecated layer

abhiskk · Aug 28, 2017 · 79edb65 · 79edb65
1 parent 34edece
commit 79edb65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_style/transformer_net.py b/neural_style/transformer_net.py
@@ -94,7 +94,7 @@ def __init__(self, in_channels, out_channels, kernel_size, stride, upsample=None
         super(UpsampleConvLayer, self).__init__()
         self.upsample = upsample
         if upsample:
-            self.upsample_layer = torch.nn.UpsamplingNearest2d(scale_factor=upsample)
+            self.upsample_layer = torch.nn.Upsample(scale_factor=upsample)
         reflection_padding = int(np.floor(kernel_size / 2))
         self.reflection_pad = nn.ReflectionPad2d(reflection_padding)
         self.conv2d = nn.Conv2d(in_channels, out_channels, kernel_size, stride)