fix batch norm warp

lhm0503 · Feb 19, 2021 · c998330 · c998330
1 parent d0b2bc5
commit c998330
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tensorlayer/app/human_pose_estimation/LCN.py b/tensorlayer/app/human_pose_estimation/LCN.py
@@ -166,9 +166,8 @@ def batch_normalization_warp(y):
     _, output_size = y.get_shape()
     output_size = int(output_size)
     out_F = int(output_size / IN_JOINTS)
-
     y = Reshape([-1, IN_JOINTS, out_F])(y)
-    y = BatchNorm(act='lrelu')(y)
+    y = BatchNorm(act='lrelu', epsilon=1e-3)(y)
     y = Reshape([-1, output_size])(y)
     return y