Skip to content

S
seminar-breakout
Commit 8543db08 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

[FasterRCNN] fix batchnorm in freeze mode

parent 1f881fcf
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tensorpack/models/batch_norm.py
View file @ 8543db08
...@@ -160,8 +160,8 @@ def BatchNorm(x, use_local_stat=None, decay=0.9, epsilon=1e-5, ...@@ -160,8 +160,8 @@ def BatchNorm(x, use_local_stat=None, decay=0.9, epsilon=1e-5,
if ctx.is_main_training_tower: if ctx.is_main_training_tower:
add_model_variable(moving_mean) add_model_variable(moving_mean)
add_model_variable(moving_var) add_model_variable(moving_var)
if use_local_stat: if ctx.is_main_training_tower and use_local_stat:
ret = update_bn_ema(xn, batch_mean, batch_var, moving_mean, moving_var, decay) ret = update_bn_ema(xn, batch_mean, batch_var, moving_mean, moving_var, decay)
else: else:
ret = tf.identity(xn, name='output') ret = tf.identity(xn, name='output')
......
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