fix: Fixed private attribute name of conv

frgfm · Jun 25, 2020 · 01687ca · 01687ca
1 parent 47668d4
commit 01687ca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/holocron/nn/modules/conv.py b/holocron/nn/modules/conv.py
@@ -76,7 +76,7 @@ def __init__(self, in_channels, out_channels, kernel_size, stride=1,
 
     def forward(self, input):
         if self.padding_mode != 'zeros':
-            return F.norm_conv2d(pad(input, self._padding_repeated_twice, mode=self.padding_mode),
+            return F.norm_conv2d(pad(input, self._reversed_padding_repeated_twice, mode=self.padding_mode),
                                  self.weight, self.bias, self.stride, _pair(0),
                                  self.dilation, self.groups, self.eps)
         return F.norm_conv2d(input, self.weight, self.bias, self.stride, self.padding,
@@ -133,7 +133,7 @@ def __init__(self, in_channels, out_channels, kernel_size, stride=1,
 
     def forward(self, input):
         if self.padding_mode != 'zeros':
-            return F.add2d(pad(input, self._padding_repeated_twice, mode=self.padding_mode),
+            return F.add2d(pad(input, self._reversed_padding_repeated_twice, mode=self.padding_mode),
                            self.weight, self.bias, self.stride, _pair(0),
                            self.dilation, self.groups, self.normalize_slices, self.eps)
         return F.add2d(input, self.weight, self.bias, self.stride, self.padding,