Как использовать оптимизатор Lazy Adam в tenorflow 2.0.0

Question

Этот код не работает: у него проблема с tf.contrib

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

Я пробовал что-то с tensorflow_addons.optimizers.LazyAdam(), но это тоже не работает.

Есть идеи, как запустить LazyAdam в tenorflow 2.0.0?

PS: только Adam работает следующим образом:

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

Answer 1

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