python - 如何在自定义训练循环中使用 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。解决了这个问题。