Merge pull request #5107 from deeplearning4j/ab_gradcheck_util

Minor fix for GradientCheckUtil method
deeplearning4j · May 9, 2018 · 9e86a0d · 9e86a0d
2 parents d80262f + f6bf576
commit 9e86a0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deeplearning4j-nn/src/main/java/org/deeplearning4j/gradientcheck/GradientCheckUtil.java b/deeplearning4j-nn/src/main/java/org/deeplearning4j/gradientcheck/GradientCheckUtil.java
@@ -351,7 +351,7 @@ public static boolean checkGradients(ComputationGraph graph, double epsilon, dou
                                          double minAbsoluteError, boolean print, boolean exitOnFirstError, INDArray[] inputs,
                                          INDArray[] labels, INDArray[] fMask, INDArray[] lMask) {
         return checkGradients(graph, epsilon, maxRelError, minAbsoluteError, print, exitOnFirstError, inputs,
-                labels, fMask, lMask);
+                labels, fMask, lMask, null);
     }
 
     public static boolean checkGradients(ComputationGraph graph, double epsilon, double maxRelError,