打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
大家好, 我用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
下图是我输出第一层卷积的不同结果。
我在GPU上使用完全相同的预训练网络、模型配置、测试代码,打印出的参数模型每次都是相同的。
热门帖子
精华帖子