卷积神经网络的主代码在哪里?

虽然我想修改卷积神经网络的代码,但我无法找到卷积神经网络的主函数。主函数是卷积和池化。

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在哪里吗?

点赞
用户1190388
用户1190388

GitHub上存在一个未解决的问题。其中一个解决方案提到升级nncunn

luarocks install nn
luarocks install cunn

另外,请查看此回复

2016-03-03 03:15:05