nvidia-digits - 从 Caffe 和 DIGITS 更新权重

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)

祝你好运！