Обновление весов от Caffe и DIGITS

Question

Я построил модель с использованием DIGITS от Nvidia, но когда я пытаюсь запустить ее с помощью caffe, я не знаю, где находятся веса. Любая идея, как я могу найти это. У меня есть архитектура, потому что это предоставляется прямо на экране модели вывода.

Answer 1

Веса недоступны ни в одной из моделей вывода, указанных в пользовательском интерфейсе Digits, однако они доступны!

Я использую 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)

Удачи!