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; }