Minor test polish

deeplearning4j · Jul 25, 2018 · 90fba44 · 90fba44
1 parent 040c93f
commit 90fba44
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...g4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/convolution/TestConvolution.java b/...g4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/convolution/TestConvolution.java
@@ -211,7 +211,8 @@ public void validateXceptionImport() throws Exception {
 
     @Test
     public void testCudnnDilation(){
-        int[] k = new int[]{2,3,4};
+        //Sanity check on dilated conv execution
+        int[] k = new int[]{2,3,4,5};
         int[] d = new int[]{1,2,3,4};
 
         for( int[] inputSize : new int[][]{{10,1,28,28}, {3,3,224,224}}) {
@@ -231,7 +232,7 @@ public void testCudnnDilation(){
                     net.init();
 
                     INDArray in = Nd4j.create(inputSize);
-                    INDArray out = net.output(in);
+                    net.output(in);
                 }
             }
         }