Обновление весов от Caffe и DIGITS - PullRequest
Новая
       45

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

0 голосов
/ 22 января 2019

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

1 Ответ

0 голосов
/ 09 мая 2019

Веса недоступны ни для одной из моделей вывода, указанных в интерфейсе 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)

Удачи!

...