Question

我已经使用Nvidia的DIGITS构建了一个模型，但是当我尝试使用caffe运行它时，我不知道权重在哪里。知道我怎么找到这个。我有体系结构，因为它是在输出模型屏幕上直接提供的。

Answer 1

从Digits UI上给出的任何输出模型都无法访问权重，但是可以访问！

我使用的是NVIDIA的DGX，可以使用python代码。为了在该平台上发挥作用（我可以在其中路由模型以保存模型，请使用以下代码：

net = caffe.Net('../models/bvlc_reference_caffenet/deploy.prototxt', 
            '../models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel', 
            caffe.TEST)
params = ['fc6', 'fc7', 'fc8']

fc_params = {pr: (net.params[pr][0].data, net.params[pr][1].data) for pr in params}

for fc in params:
print '{} weights are {} dimensional and biases are {} dimensional'.format(fc, fc_params[fc][0].shape, fc_params[fc][1].shape)

祝你好运！

从Caffe和DIGITS更新权重

1 个答案: