首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >关于tensorflow将训练的网络结构转成.pb文件再进行加载时候的报错?

关于tensorflow将训练的网络结构转成.pb文件再进行加载时候的报错?

提问于 2018-05-31 11:06:06
回答 4关注 2查看 2.5K

graph_def is invalid at node 'bn1/Assign': Input tensor 'bn1/bn1_moving_mean:0' Cannot convert a tensor of type float32 to an input of type float32_ref 来这里向各位大神请教,这是在加载使用tensorflow生成的.pb文件时候产生的,然后由于网络结构中有bn层,在加载时候就会报错。发现几乎没有中文评论具体说这个问题,找到的修改方法由于也没有人具体说下原理根据,不太会活用。下面是我找的别人的修改方法,然后出现了"RefSwitch"这些应该是使用了tf.cond(),然后我修改了tf.cond(),不要这个函数,依然报错,因为再bn层具体实现中,我使用了tf.Assign()操作,不知道是不是这个原因导致了本文段开头贴出来的那些错误

for node in input_graph_def.node:

if node.op == 'RefSwitch':

node.op = 'Switch'

for index in xrange(len(node.input)):

if 'moving_' in node.input[index]:

node.input[index] = node.input[index] + '/read'

elif node.op == 'AssignSub':

node.op = 'Sub'

if 'use_locking' in node.attr: del node.attr['use_locking']

回答

和开发者交流更多问题细节吧,去 写回答
相关文章

相似问题

相关问答用户
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档