ResNet c++中加载问题

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

背景:

resnet模型固化方法:

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

问题:

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

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

你可能感兴趣的:(神经网络,Python)