tensorflow调试的一种方式

由于tensorflow采用构建图,在会话中再运行图的方式,使得调试非常麻烦。尤其是想获取网络中间某个tensor的时候。对此,可以采用以下方法

在构建完成网络所有结构之后,G = tf.get_default_graph()获取图,然后使用tensr = G.get_tensor_by_name(“TensorName:0”)的方式获取想要的tensor,再sess.run该tensor就可以获取它的值了。TensorName可以通过查看网络ckpt、pb文件的方式获取。

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注