torch / nngraph

Graph Computation for nn

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ittermediate feature maps

varghesealex90 opened this issue · comments

Hi;
I made a very simple convolution network using nngraph. I would like to see the intermediate convolution features
require 'nngraphs'

a=nn.Identity() ()
b=nn.SpatialConvolution(3,32,5,5)(a)
c= nn.SpatialConvolution(32,4,5,5)(b)

model= nn.Sequential()
model:add(nn.gModule({a},{c}))

Now I would like to see the feature map from the first conv.

Please help