tensorflow恢复restore官网ckpt出错,Tensor name not found

搭建好vgg,resnet等网络,加载tensorflow官网(https://github.com/tensorflow/models/tree/master/research/slim)提供的预训练ckpt模型.
使用restore运行出错,一些节点(conv/biases)在ckpt中不存在,出错无法运行.正确的做法,
slim.assign_from_checkpoint_fn(’./data/pre_trained/resnet_v1_50.ckpt’,
slim.get_trainable_variables(),
ignore_missing_vars=True)

with tf.Session() as sess:
sess.run(init)
# resotre 预训练模型
variable_restore_op(sess)

运行会提醒丢失的,无法加载的模型参数,程序正常运行.

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值