Commit aa673e97 authored by Yuxin Wu's avatar Yuxin Wu

bump TF required version

parent 3f9fa67f
......@@ -62,7 +62,7 @@ You can also define your own trainer for non-standard training (e.g. GAN).
Dependencies:
+ Python 2 or 3
+ TensorFlow >= 1.0.0rc1
+ TensorFlow >= 1.0.0
+ Python bindings for OpenCV
```
pip install --user -U git+https://github.com/ppwwyyxx/tensorpack.git
......
......@@ -124,6 +124,8 @@ def update_bn_ema(xn, batch_mean, batch_var, moving_mean, moving_var, decay):
name='var_ema_op')
add_model_variable(moving_mean)
add_model_variable(moving_var)
# seems faster than delayed update, but might behave otherwise in distributed settings.
with tf.control_dependencies([update_op1, update_op2]):
return 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