gpt4 book ai didi

python - 来自 model.ckpt.meta 文件的 Tensorflow View 图

转载 作者:行者123 更新时间:2023-12-01 07:47:50 26 4
gpt4 key购买 nike

我有一个 model.ckpt.meta 文件,我只想查看架构/图形结构。我无法从 model.ckpt.meta 文件中找到如何执行此操作。

感谢米兰遵循代码:

tf.train.import_meta_graph("./model.ckpt.meta")
for n in tf.get_default_graph().as_graph_def().node:
print(n)
with tf.Session() as sess:
writer = tf.summary.FileWriter("./output/", sess.graph)
writer.close()

我得到了下面的图表。但很多架构都缺失了。 enter image description here

编辑:糟糕!我忘记双击模型了。已排序。

最佳答案

可以使用tf.train.import_meta_graph在python中导入元图,然后遍历图中的节点,例如:

import tensorflow as tf

tf.train.import_meta_graph("./model.ckpt-200000.meta")
for n in tf.get_default_graph().as_graph_def().node:
print(n)

导入后,您可以为 Tensorboard 制作一个摘要文件,这样您就可以很好地可视化图形:

with tf.Session() as sess:
writer = tf.summary.FileWriter("./output/", sess.graph)
writer.close()

要查看 Tensorboard 中保存的摘要文件,请运行 tensorboard --logdir=./output/

关于python - 来自 model.ckpt.meta 文件的 Tensorflow View 图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56363333/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com