Commit 71e106fe authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #1525 from toxic-0518/patch-1

Update resnet_model.py
parents 62b33958 0ce70bdd
...@@ -185,7 +185,7 @@ class ResNet(object): ...@@ -185,7 +185,7 @@ class ResNet(object):
trainable=False) trainable=False)
tf.summary.histogram(mean.op.name, mean) tf.summary.histogram(mean.op.name, mean)
tf.summary.histogram(variance.op.name, variance) tf.summary.histogram(variance.op.name, variance)
# elipson used to be 1e-5. Maybe 0.001 solves NaN problem in deeper net. # epsilon used to be 1e-5. Maybe 0.001 solves NaN problem in deeper net.
y = tf.nn.batch_normalization( y = tf.nn.batch_normalization(
x, mean, variance, beta, gamma, 0.001) x, mean, variance, beta, gamma, 0.001)
y.set_shape(x.get_shape()) y.set_shape(x.get_shape())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment