Merge branch 'nn_fast_reset'

1 files changed, 11 insertions, 6 deletions

diff --git a/SpatialConvolutionMM.lua b/SpatialConvolutionMM.lua
index aa358e5..cc6a718 100644
--- a/SpatialConvolutionMM.lua
+++ b/SpatialConvolutionMM.lua
@@ -25,12 +25,17 @@ function SpatialConvolutionMM:reset(stdv)
    else
       stdv = 1/math.sqrt(self.kW*self.kH*self.nInputPlane)
    end
-   self.weight:apply(function()
-                        return torch.uniform(-stdv, stdv)
-                     end)
-   self.bias:apply(function()
-                      return torch.uniform(-stdv, stdv)
-                   end)   
+   if nn.oldSeed then
+      self.weight:apply(function()
+         return torch.uniform(-stdv, stdv)
+      end)
+      self.bias:apply(function()
+         return torch.uniform(-stdv, stdv)
+      end)  
+   else
+      self.weight:uniform(-stdv, stdv)
+      self.bias:uniform(-stdv, stdv)
+   end
 end
 
 function SpatialConvolutionMM:updateOutput(input)