如何在 tensorflow 2.0.0 中使用 Lazy Adam 优化器

Question

此代码无效：tf.contrib

有问题

model.compile(optimizer=TFOptimizer(tf.contrib.opt.LazyAdamOptimizer()), loss='categorical_crossentropy')

我已经用 tensorflow_addons.optimizers.LazyAdam() 尝试过一些方法，但这也不起作用。

关于如何在 tensorflow 2.0.0 中 运行 LazyAdam 有什么想法吗？

PS：只有 Adam 效果很好，如下所示：

model.compile(optimizer=tf.keras.optimizers.Adam(), loss='categorical_crossentropy')

Answer 1

import tensorflow_addons as tfa
optimizer = tfa.optimizers.LazyAdam()

tensorflow_addons 是 TensorFlow 2.x 的额外功能，但现在 Tensorflow 2.x 仍然不是很稳定，如果你正面临 module 'tensorflow_core.keras.utils' has no attribute 'register_keras_serializable'，请尝试更新你tensorflow到最新的稳定版。