在张量流的每个时期之后如何维护EMA预测？

Question

我必须查看每个时期之后基于指数移动平均值的神经网络的性能。目前，我正在这样做：

with tf.Session as sess:
    train(sess):
        while i in epochs():
            model_build(reuse = True, getter=None)
            run_epoch(updates_ema_as_well)
            model_build(reuse = True, getter = custom_getter_that_returns_EMA_for_each_variable)
            model.predict(validation_data)

问题是，我认为每次调用model_build()时，它都会构建多个图形。我不熟悉tensorflow来正确处理这个问题。请建议我改善自己的方法