从元图中检索变量名

Question

我目前正在尝试从另一个项目( vizdoom平台上的https://github.com/pathak22/noreward-rl)检索经过训练的TF模型。

我已经成功地将模型导入到我的新项目中，方法是：

session = tf.Session()
print("Loading model from: ", model_savefile)
saver = tf.train.import_meta_graph(model_savefile + '.meta')
saver.restore(session, model_savefile)

但是，我无法访问生成此存储文件的代码(我认为它是通过OpenAI Gym实现的，但不确定)，因此我不知道应该使用哪些名称来提供我的输入。

你知道怎么做吗？

提前谢谢你

Answer 1

导入MetaGraph会将操作添加到默认图形中。

打印图表中的所有操作：

print(tf.get_default_graph().get_operations())

打印类似以下内容：

[<tf.Operation 'Placeholder' type=Placeholder>, <tf.Operation 'mul/y' type=Const>, <tf.Operation 'mul' type=Mul>]

仅打印占位符：

print([op for op in tf.get_default_graph().get_operations() if op.type == 'Placeholder'])

打印类似以下内容：

[<tf.Operation 'Placeholder' type=Placeholder>]