发新帖

关于 output_node_names

[复制链接]
1502 9

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
训练的模型要应用于手机之中,目前是使用TensorFlowMobile方案,生成pb文件。
看了官方的文档中说要定义 output_node_names ,


可是我训练的时候没有对输出节点定义name ,是不是意味着我要重新跑一遍训练?





yunhai_luo已获得悬赏 10 金币+5 金币

最佳答案

简单提一个小问题,解决你这个问题(“拯救”已经训练好的模型而不用重头再来)的具体细节可能也与你当初是如何保存模型的有关。如果你是用summary_op保存的模型,那么应该有tfevents文件,也应该可以用tensorboar ...
本楼点评(0) 收起

精彩评论9

feiluosb  TF荚荚  发表于 2018-4-25 11:14:26 来自手机  | 显示全部楼层
用tensorboard看整个图结构 找输出点
本楼点评(0) 收起
tensorfyx  TF豆豆  发表于 2018-4-25 15:14:15 | 显示全部楼层
feiluosb 发表于 2018-4-25 11:14
用tensorboard看整个图结构 找输出点

求详细方案,本人小白一个。
首先我对这个 output_node_names没有真正理解,是要设置输出层节点的名字,还是输出结果的变量名?




本楼点评(0) 收起
舟3332  TF芽芽  发表于 2018-4-25 16:08:47 来自手机  | 显示全部楼层
tensorfyx 发表于 2018-4-25 15:14
求详细方案,本人小白一个。
首先我对这个 output_node_names没有真正理解,是要设置输出层节点的名字, ...

节点有默认的名字的。可以看一下论坛里的另一篇我问的帖子,用 tensorboard 查看
本楼点评(0) 收起
tensorfyx  TF豆豆  发表于 2018-4-25 18:32:34 | 显示全部楼层
舟3332 发表于 2018-4-25 16:08
节点有默认的名字的。可以看一下论坛里的另一篇我问的帖子,用 tensorboard 查看 ...

帮忙帮到底,请给个链接,刚去你主页翻了一圈没找到。
本楼点评(0) 收起
舟3332  TF芽芽  发表于 2018-4-26 21:51:30 来自手机  | 显示全部楼层
tensorfyx 发表于 2018-4-25 18:32
帮忙帮到底,请给个链接,刚去你主页翻了一圈没找到。

https://www.tensorflowers.cn/t/326

可以参考下这个
本楼点评(0) 收起
M丶Sulayman  TF豆豆  发表于 2018-4-26 21:55:19 | 显示全部楼层
路过,默默学习一下
本楼点评(0) 收起
yunhai_luo  TF豆豆  发表于 2018-4-27 01:01:37 来自手机  | 显示全部楼层
tensorfyx 发表于 2018-4-25 18:32
帮忙帮到底,请给个链接,刚去你主页翻了一圈没找到。

简单提一个小问题,解决你这个问题(“拯救”已经训练好的模型而不用重头再来)的具体细节可能也与你当初是如何保存模型的有关。如果你是用summary_op保存的模型,那么应该有tfevents文件,也应该可以用tensorboard --logdir=/path/to/tfevents看计算图找到输出节点及节点名。但是如果你只是tf.train.Saver保存,应该不会有tfevents文件,也就不能用tensorboard。这时候你可能应该考虑写一个新的脚本用tf.train.import_meta_graph导入MataGraph(model.ckpt.meta),恢复模型计算图(应该无需重新训练模型),然后用tensorboard或者其他方法找出输出节点。
本楼点评(3) 收起
  • tensorfyx感谢,这个问题已经解决,找到了输出节点,修改了代码命名。我们restore已经保存的模型,然后恢复计算图,再重新保存为pb文件就行了。
    2018-4-28 09:02 回复
  • yunhai_luo多谢反馈,希望有帮上忙。
    2018-4-28 11:53 回复
  • tensorfyx回复 yunhai_luo :提供的思路帮助很大
    2018-4-28 15:01 回复
tensorfyx  TF豆豆  发表于 2018-7-3 17:34:32 | 显示全部楼层
感谢,这个问题已经解决,找到了输出节点,修改了代码命名。我们restore已经保存的模型,然后恢复计算图,再重新保存为pb文件就行了。
本楼点评(0) 收起
neverchange  TF豆豆  发表于 2018-7-4 00:41:12 | 显示全部楼层
tensorboard里可以看得到
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

快速回复 返回顶部 返回列表