Как извлечь данные о весе из слоя Convolution, размер группы которого равен двум? - PullRequest
Новая
       82

Как извлечь данные о весе из слоя Convolution, размер группы которого равен двум?

0 голосов
/ 16 апреля 2020

Как я знаю, Ale xnet имеет три слоя, размер группы которых равен 2. [conv1, conv2 , conv3, conv4 , conv5 ]

Я пытался получить весовые параметры из conv2, conv4, conv5, используя net.params[layer_name][0].data

Так что я ожидал, что каждые три слоя дают мне 2 вида массива весов, потому что его размер группы равен двум, но я получил только один. 

for name in [conv1,conv2,conv3,conv4,conv5]:
    print net.params[name][0].data.shape
----result----- 
(96,3,11,11)
(256,48,5,5) # this is first weight of groups where is the other one?
(385,256,3,3)
(384,192,3,3)
(256,192,3,3)

если размер группы равен двум, я думаю, что слой 'conv2' должен иметь два (256,48,5,5) массива массива.

...