model.add(tf.keras.layers.BatchNormalization())