似乎当我使用nngraph模型时,很容易这样做。 如果我只使用普通转换器,那么我该怎么做呢?
答案 0 :(得分:0)
定义模型时,例如带有nn.Sequential()
的convnet并向其添加模块
net = nn.Sequential()
net:add(nn.SpatialConvolution(3,3,1,1)
net:add(...) -- add other modules
您可以访问具有net.modules[n]
的模块(n
是模块的索引,使用print(net)
查看整个网络及其模块)。然后每个模块必须声明变量output
和gradInput
(模块相对于其输入的渐变),然后您可以使用
net.modules[n].output