虽然我想修改卷积网络的代码,但我找不到卷积网络的主程序。主要程序是卷积和汇集。
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
在哪里吗?
答案 0 :(得分:1)
这是一个未解决的问题on GitHub。其中一个the solution提及升级nn
和cunn
:
luarocks install nn
luarocks install cunn
另外,请查看this reply。