×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

请问为什么每次加载模型后,参数都不一样? 已解决 Gummary2020-07-03 13:48:28 回复 3 查看
请问为什么每次加载模型后,参数都不一样?
分享到:

大家好, 我用pycaffe加载预训练模型之后,想看看模型参数加载是否正确,但是每次打印模型的参数都不一致,请问这是为什么?


我的代码如下:

import caffe 

prototxt="net.prototxt"
weights="pretrain.caffemodel"

caffe.set_mode_mlu()

net = caffe.Net(prototxt, weights, caffe.TEST)

print(net.params['conv1'][0].data)

使用的网络为lenet


下图是我输出第一层卷积的不同结果。

image.png


image.png


我在GPU上使用完全相同的预训练网络、模型配置、测试代码,打印出的参数模型每次都是相同的。

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭