Skip to content

Commit c998330

Browse files
committed
fix batch norm warp
1 parent d0b2bc5 commit c998330

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

  • tensorlayer/app/human_pose_estimation

tensorlayer/app/human_pose_estimation/LCN.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,9 +166,8 @@ def batch_normalization_warp(y):
166166
_, output_size = y.get_shape()
167167
output_size = int(output_size)
168168
out_F = int(output_size / IN_JOINTS)
169-
170169
y = Reshape([-1, IN_JOINTS, out_F])(y)
171-
y = BatchNorm(act='lrelu')(y)
170+
y = BatchNorm(act='lrelu', epsilon=1e-3)(y)
172171
y = Reshape([-1, output_size])(y)
173172
return y
174173

0 commit comments

Comments
 (0)