2017-05-17 2 views

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