From 02cb2ad9f0926c26e1a9e7988754feb95b293654 Mon Sep 17 00:00:00 2001 From: MultiLi Date: Fri, 14 Jul 2017 16:56:23 -0700 Subject: [PATCH] Get rid of annoying parameters of BatchNorm layer --- python/mxnet/visualization.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/python/mxnet/visualization.py b/python/mxnet/visualization.py index 6e56dfb612ec..10a24241c730 100644 --- a/python/mxnet/visualization.py +++ b/python/mxnet/visualization.py @@ -240,6 +240,14 @@ def looks_like_weight(name): return True if name.endswith("_bias"): return True + if name.endswith("_var"): + return True + if name.endswith("_beta"): + return True + if name.endswith("_gamma"): + return True + if name.endswith("_mean"): + return True return False # make nodes