如何在自定义训练循环中使用tf.keras.layers.BatchNormalization（）？

Question

一段时间后，我回到了tensorflow，看来情况已经完全改变了。

但是，以前我曾经在训练循环中使用tf.contrib....batch_normalization和以下内容：

update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS)
with tf.control_dependencies(update_ops):
    train_op = optimizer.minimize(cnn.loss, global_step=global_step)

但是似乎找不到contrib，并且tf.keras.layers.BatchNormalization的工作方式也不相同。另外，我在他们的documentation中找不到任何培训说明。

因此，感谢您提供任何帮助信息。

Answer 1

我开始使用pyTorch。解决了这个问题。