卷积网的主要代码在哪里?

时间:2016-03-03 01:50:37

标签: lua deep-learning torch

虽然我想修改卷积网络的代码,但我找不到卷积网络的主程序。主要程序是卷积和汇集。

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

1 个答案:

答案 0 :(得分:1)

这是一个未解决的问题on GitHub。其中一个the solution提及升级nncunn

luarocks install nn
luarocks install cunn

另外,请查看this reply