adding back assertion until a proper backward is added

stephenrawls · Apr 14, 2016 · 7111bce · 7111bce
1 parent 9cffea5
commit 7111bce
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/BatchNormalization.lua b/BatchNormalization.lua
@@ -125,6 +125,7 @@ end
 local function backward(self, input, gradOutput, scale, gradInput, gradWeight, gradBias)
    self:checkInputDim(input)
    self:checkInputDim(gradOutput)
+   assert(self.train == true, 'should be in training mode when self.train is true')
    assert(self.save_mean and self.save_std, 'must call :updateOutput() first')
 
    input, gradOutput = makeContiguous(self, input, gradOutput)