From cdde99ad3f71720d75f663d6b142a7894bda9308 Mon Sep 17 00:00:00 2001 From: Lyken17 Date: Wed, 31 May 2017 22:38:53 -0700 Subject: [PATCH] hide _move_var _move_mean _beta _gamma for visualization --- python/mxnet/visualization.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/mxnet/visualization.py b/python/mxnet/visualization.py index 6e56dfb612ec..97b6bfa25b1b 100644 --- a/python/mxnet/visualization.py +++ b/python/mxnet/visualization.py @@ -240,6 +240,9 @@ def looks_like_weight(name): return True if name.endswith("_bias"): return True + if name.endswith("_beta") or name.endswith("_gamma") or \ + name.endswith("_moving_var") or name.endswith("_moving_mean"): + return True return False # make nodes