fix in conv repr

760chong · Dec 29, 2016 · 239ae94 · 239ae94
1 parent 55e850d
commit 239ae94
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch/nn/modules/conv.py b/torch/nn/modules/conv.py
@@ -253,8 +253,8 @@ def __repr__(self):
         padding_str = ', padding=(' + str(padt) \
                       + ', ' + str(padh) + ', ' + str(padw) + ')' \
                       if padt != 0 and padh != 0 and padw !=0 else ''
-        return  self.__class__.__name__ + ' (' + str(in_channels) \
-            + ' -> ' + str(out_channels) \
+        return  self.__class__.__name__ + ' (' + str(self.in_channels) \
+            + ' -> ' + str(self.out_channels) \
             + ', size=(' + str(kt) + ', ' + str(kh) + ', ' + str(kw) + ')' \
             + ', stride=(' + str(dt) + ', ' + str(dh) + ', ' + str(dw) + ')' \
             + padding_str + ')'