Don't you have to update batch norm statistics?
sh1ng opened this issue · comments
Vladimir commented
Hi,
thank you for the implementation!
I can't find update stat operation
from the documentation
update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS) with tf.control_dependencies(update_ops): train_op = optimizer.minimize(loss)
https://www.tensorflow.org/api_docs/python/tf/contrib/layers/batch_norm