从Caffe和DIGITS更新权重

Question

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

Answer 1

权重不能从Digits UI上给出的任何输出模型中访问，但是它们可以访问！

我使用的是NVIDIAs 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)

祝好运!