message: Add batch norm updates for estimator.
diff: @@ -55,6 +55,7 @@ def model_fn(features, labels, mode, params):
  learning_rate, momentum=config.momentum)
  logits, saved_hidden = model(inputs, training=True)
  grads, loss = model.compute_gradients(saved_hidden, labels, training=True)
  + with tf.control_dependencies(model.get_updates_for(inputs)):
  train_op = optimizer.apply_gradients(
  zip(grads, model.trainable_variables), global_step=global_step)
  