如何从恢复的张量流图获取优化器对象？

Question

我使用

使用adam-optimizer创建了一个张量流模型

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上也讨论了类似的问题。