什么可以用来替换 TensorFlow 2 中的 tf.train.GradientDescentOptimizer

Question

我正在尝试将我的 TensorFlow 1.4 代码转换为 TensorFlow 2，但tf.train.GradientDescentOptimizer在 TF 2 中没有更多代码了。

tf.train.GradientDescentOptimizer 的替代品应该是什么？

我发现了同样的东西，tf.compat.v1.train.但我不应该使用它，因为这些东西tf.compat迟早会被删除。

Answer 1

您可以将其替换为，在此处tf.keras.optimizers.SGD()定义。 这里（跳到第三点）是官方消息，其中 TF 团队提到使用这个 keras 优化器。