diff options
author | Gregory Chanan <gchanan@fb.com> | 2016-11-30 20:58:51 +0300 |
---|---|---|
committer | Gregory Chanan <gchanan@fb.com> | 2016-12-01 20:28:20 +0300 |
commit | 792f46fcf3c1d1304f32bd3662a1af68a7216026 (patch) | |
tree | f70dc2b0b53a5697b4eb6110008800f69b39ff25 | |
parent | 05890b643148f4169302345dfa885547ad4d8ee5 (diff) |
Remove resizing of output in lua.
-rw-r--r-- | BatchNormalization.lua | 1 | ||||
-rw-r--r-- | SpatialUpSamplingBilinear.lua | 1 | ||||
-rw-r--r-- | SpatialUpSamplingNearest.lua | 7 |
3 files changed, 0 insertions, 9 deletions
diff --git a/BatchNormalization.lua b/BatchNormalization.lua index 1cd30aa..8dfc576 100644 --- a/BatchNormalization.lua +++ b/BatchNormalization.lua @@ -116,7 +116,6 @@ function BN:updateOutput(input) input = makeContiguous(self, input) input = makeBatch(self, input) - self.output:resizeAs(input) self.save_mean = self.save_mean or input.new() self.save_mean:resizeAs(self.running_mean) self.save_std = self.save_std or input.new() diff --git a/SpatialUpSamplingBilinear.lua b/SpatialUpSamplingBilinear.lua index 8f19f91..e86beb3 100644 --- a/SpatialUpSamplingBilinear.lua +++ b/SpatialUpSamplingBilinear.lua @@ -79,7 +79,6 @@ function SpatialUpSamplingBilinear:updateOutput(input) local xdim = input:dim() local ydim = xdim - 1 self:setSize(input) - self.output:resize(self.outputSize) input.THNN.SpatialUpSamplingBilinear_updateOutput( input:cdata(), self.output:cdata(), diff --git a/SpatialUpSamplingNearest.lua b/SpatialUpSamplingNearest.lua index b1b261a..362ae73 100644 --- a/SpatialUpSamplingNearest.lua +++ b/SpatialUpSamplingNearest.lua @@ -39,13 +39,6 @@ function SpatialUpSamplingNearest:updateOutput(input) end self.outputSize[ydim] = self.outputSize[ydim] * self.scale_factor self.outputSize[xdim] = self.outputSize[xdim] * self.scale_factor - -- Resize the output if needed - if input:dim() == 3 then - self.output:resize(self.outputSize[1], self.outputSize[2], - self.outputSize[3]) - else - self.output:resize(self.outputSize) - end input.THNN.SpatialUpSamplingNearest_updateOutput( input:cdata(), self.output:cdata(), |