Added negative inf check

SeanNaren · Dec 2, 2016 · e453440 · e453440
1 parent 2975947
commit e453440
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Network.lua b/Network.lua
@@ -130,7 +130,7 @@ function Network:trainNetwork(epochs, optimizerParams)
         sizes = self.calSize(sizes)
         local predictions = self.model:forward(inputs)
         local loss = criterion:forward(predictions, targets, sizes)
-        if loss == math.huge then loss = 0 print("Recieved an inf cost!") end
+        if loss == math.huge or loss == -math.huge then loss = 0 print("Recieved an inf cost!") end
         self.model:zeroGradParameters()
         local gradOutput = criterion:backward(predictions, targets)
         self.model:backward(inputs, gradOutput)