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

Question

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

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

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

Answer 1

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