【问题标题】:Where is the main code of convolutional nets?卷积网络的主要代码在哪里?
【发布时间】: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在哪里?

【问题讨论】:

标签: lua deep-learning torch


【解决方案1】:

同样有一个未解决的问题on GitHubthe solution 之一提到升级 nncunn

luarocks install nn
luarocks install cunn

另外,看看this reply

【讨论】:

  • 太棒了!非常感谢!
猜你喜欢
  • 2018-04-06
  • 1970-01-01
  • 2016-08-08
  • 2018-06-15
  • 2017-01-01
  • 1970-01-01
  • 2015-04-09
  • 2019-08-18
相关资源
最近更新 更多