Skip to content
This repository was archived by the owner on Dec 19, 2023. It is now read-only.

Commit c998330

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

File tree

1 file changed

+1
-2
lines changed
  • tensorlayer/app/human_pose_estimation

1 file changed

+1
-2
lines changed

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)