Il semble que lorsque j'utilise le modèle nngraph, il est facile de le faire. Si j'utilise uniquement la conversion normale, comment puis-je le faire?Comment obtenir la sortie des couches intermédiaires dans Torch7?
0
A
Répondre
0
Lorsque vous définissez un modèle, par exemple un convnet avec nn.Sequential()
et en ajoutant des modules à comme
net = nn.Sequential()
net:add(nn.SpatialConvolution(3,3,1,1)
net:add(...) -- add other modules
vous pouvez accéder à un module avec net.modules[n]
(n
est l'indice du module, utilisez print(net)
pour voir votre réseau entier et ses modules). Ensuite, chaque module doit préciser les variables output
et gradInput
(gradient du module par rapport à son entrée), vous pouvez accéder à la sortie du n-ième couche intermédiaire avec
net.modules[n].output