如何用TensorFlow实现打印模型每一层的输出结果?
- 内容介绍
- 文章标签
- 相关推荐
本文共计301个文字,预计阅读时间需要2分钟。
在test.py中,可以通过以下代码直接生成带weight的pb文件,也可以使用tf官方的freeze_graph.py将ckpt转换为pb文件。
pythonconstant_graph=graph_util.convert_variables_to_constants(sess, sess.graph_def, ['net_loss/inference/e'])
在test.py中可以通过如下代码直接生成带weight的pb文件,也可以通过tf官方的freeze_graph.py将ckpt转为pb文件。
本文共计301个文字,预计阅读时间需要2分钟。
在test.py中,可以通过以下代码直接生成带weight的pb文件,也可以使用tf官方的freeze_graph.py将ckpt转换为pb文件。
pythonconstant_graph=graph_util.convert_variables_to_constants(sess, sess.graph_def, ['net_loss/inference/e'])
在test.py中可以通过如下代码直接生成带weight的pb文件,也可以通过tf官方的freeze_graph.py将ckpt转为pb文件。

