如何从恢复的tensorflow图中获取优化器对象？

Question

我使用adam优化器创建了一个tensorflow模型

tf.train.AdamOptimizer(..., name='AO').minimize(...)

在我存储模型并从模型文件中还原它之后，我想使用优化器最小化一个不同的变量集。我的问题是如何获得名为'AO'的优化器对象

Answer 1

如果您的优化器是这样使用的：

optimizer = tf.train.AdamOptimizer(1e-4)
train_step = optimizer.minimize(loss)

调用时，优化器状态默认保存：

saver = tf.train.Saver().save()

恢复如下：

saver = tf.train.import_meta_graph(modelfile+ '.meta')
saver.restore(sess, tf.train.latest_checkpoint('./'))
train_step = tf.get_collection("train_step")[0]

有关保存检查点（here）的详细信息 在github和SO上也讨论了类似的问题