From 7946c3961a39120c13f1bacaa137f5592e49ab66 Mon Sep 17 00:00:00 2001 From: JackieWu Date: Sun, 26 Aug 2018 03:40:37 +0800 Subject: [PATCH] add activation information for mxnet.gluon.nn._Conv (#12354) --- python/mxnet/gluon/nn/conv_layers.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/mxnet/gluon/nn/conv_layers.py b/python/mxnet/gluon/nn/conv_layers.py index 96ecc21c81b3..5f20d20c02ab 100644 --- 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__,