【发布时间】:2016-06-16 03:47:59
【问题描述】:
虽然我想修改卷积网络的代码,但我找不到卷积网络的主程序。主要的套路是卷积和池化。
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在哪里?
【问题讨论】:
-
nn包的底层实现在这里; github.com/torch/nn/tree/master/lib/THNN/generic。在我的手电筒文件夹中,它们位于目录
~/torch/extra/nn/generic。
标签: lua deep-learning torch