From 3e92b74d23559a2b73e083a8bad85b21514f2fc2 Mon Sep 17 00:00:00 2001 From: Tom Manion Date: Sat, 30 Dec 2017 16:18:17 -0600 Subject: [PATCH] Do not let dy be NaN --- src/convnet_layers_loss.js | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/convnet_layers_loss.js b/src/convnet_layers_loss.js index 5906e34b..e9303a35 100644 --- a/src/convnet_layers_loss.js +++ b/src/convnet_layers_loss.js @@ -137,6 +137,9 @@ var i = y.dim; var yi = y.val; var dy = x.w[i] - yi; + if (isNaN(dy)) { + dy = 0; + } x.dw[i] = dy; loss += 0.5*dy*dy; }