Update train.py

train.py

@@ -7,7 +7,8 @@
 optimizer=tf.train.AdamOptimizer(cfg.learning_rate)
 #optimizer = tf.train.MomentumOptimizer(learning_rate=cfg.learning_rate, momentum=cfg.momentum, use_nesterov=True)
 update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS)
-train_op=optimizer.minimize(loss)
+with tf.control_dependencies(update_ops):
+    train_op=optimizer.minimize(loss)
 var_list = tf.trainable_variables()
 g_list = tf.global_variables()
 bn_moving_vars = [g for g in g_list if 'moving_mean' in g.name]