add activation information for mxnet.gluon.nn._Conv (apache#12354)

anirudh2290 · Sep 19, 2018 · 7946c39 · 7946c39
1 parent 420d222
commit 7946c39
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/mxnet/gluon/nn/conv_layers.py b/python/mxnet/gluon/nn/conv_layers.py
@@ -153,6 +153,8 @@ def __repr__(self):
             s += ', groups={num_group}'
         if self.bias is None:
             s += ', bias=False'
+        if self.act:
+            s += ', {}'.format(self.act)
         s += ')'
         shape = self.weight.shape
         return s.format(name=self.__class__.__name__,