Изменения

import tensorflow as tf  # ...  is_train = tf.placeholder(tf.bool, name="is_train");  # ...  x_norm = tf.layers.batch_normalization(x, training=is_train)  # ...  update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS) with tf.control_dependencies(update_ops): train_op = optimizer.minimize(loss)