python - TensorFlow MovingAverageOptimizer 在 V1 和 V2 之间的兼容性

Question

我有以下最初用 TensorFlow 1.X 编写的优化器

optimizer = tf.train.AdamOptimizer(AdamOptimizer()
optimizer = tf.contrib.opt.MovingAverageOptimizer(optimizer, average_decay=moving_average_decay)

但现在我被迫使用 Tensorflow 2.2，我尝试迁移所有内容，以便它支持 v1 的所有功能。我设法“修复”了 AdamOptimizer：

optimizer = tf.compat.v1.train.AdamOptimizer()

但我找不到 MovingAverage() 的任何实现。有任何想法吗？

标签： pythonpython-3.xtensorflowtensorflow2.0moving-average