卷积网络的主要代码在哪里？

Question

虽然想修改卷积网络的代码，但是找不到卷积网络的主程序。主要套路就是卷积和池化

SpatialConvolution.lua 有下面的代码。

 96 function SpatialConvolution:updateOutput(input)
 97    backCompatibility(self)
 98    viewWeight(self)
 99    input = makeContiguous(self, input)
100    local out = input.nn.SpatialConvolutionMM_updateOutput(self, input) -- where?
101    unviewWeight(self)
102    return out
103 end

所以我认为例程在 SpatialConvolutionMM 中。但是，SpatialConvolutionMM.lua 有下面的代码。

 65 function SpatialConvolutionMM:updateOutput(input)
 66    -- backward compatibility
 67    if self.padding then
 68       self.padW = self.padding
 69       self.padH = self.padding
 70       self.padding = nil
 71    end
 72    input = makeContiguous(self, input)
 73    return input.nn.SpatialConvolutionMM_updateOutput(self, input) -- where??
 74 end

那么有人知道SpatialConvolutionMM_updateOutput在哪里吗？

Answer 1

有一个未解决的问题 on GitHub for the same. One of the solution 提到升级 nn 和 cunn:

luarocks install nn
luarocks install cunn

另外，看看 this reply。

卷积网络的主要代码在哪里？

Where is the main code of convolutional nets?

lua

deep-learning

torch