Merge pull request apache#19 from MultiLi/master

Get rid of annoying parameters of BatchNorm layer
StOnEGiggity · Jul 14, 2017 · b90b9b5 · b90b9b5
2 parents 45801ef + 02cb2ad
commit b90b9b5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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