ResNet c++中加载问题

resne模型使用 python加载和C++加载结果不一样。

背景:

resnet模型固化方法:

可以推广到所有包含BatchNorm的模型
首先,训练一个resnet,使用tf.train.saver分别保存结构、权重;
然后,定义一个测试模型,加载权重,固化为pb模型;

问题:

刚开始是因为训练时除了训练模型外,还有测试模型,但是固化的时候仅定义了测试模型,导致加载不对,固化出错,后面发现问题了,定义了测试模型跟训练模型,然后再测试就对了。
注:想要固化模型不知道输出的节点名称时,可以直接打印该节点。

以上,均是在python中的情况,在C++中就不对,就算正确训练的模型也无法得到和python一样的结果;
但是换成 简单的CNN两种语言的结果就又一样了,非常费解。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值