diff --git a/deeplearning4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/ValidateCuDNN.java b/deeplearning4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/ValidateCuDNN.java index a9e4d185f..1d393aaf4 100644 --- a/deeplearning4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/ValidateCuDNN.java +++ b/deeplearning4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/ValidateCuDNN.java @@ -248,7 +248,7 @@ public class ValidateCuDNN extends BaseDL4JTest { Nd4j.getRandom().setSeed(12345); INDArray features = Nd4j.rand(fShape); INDArray labels = Nd4j.rand(lShape); - Nd4j.getExecutioner().exec(new IsMax(labels, 1)); + labels = Nd4j.exec(new IsMax(labels, 1)); List testCaseList = new ArrayList<>(); @@ -256,7 +256,7 @@ public class ValidateCuDNN extends BaseDL4JTest { for (int i = 0; i < 6; i++) { INDArray f = Nd4j.rand(fShape); INDArray l = Nd4j.rand(lShape); - Nd4j.getExecutioner().exec(new IsMax(l, 1)); + Nd4j.exec(new IsMax(l, 1))[0]; dataSets.add(new DataSet(f, l)); } DataSetIterator iter = new ExistingDataSetIterator(dataSets); diff --git a/deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/layers/objdetect/Yolo2OutputLayer.java b/deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/layers/objdetect/Yolo2OutputLayer.java index 2610603bc..bb551cd3f 100644 --- a/deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/layers/objdetect/Yolo2OutputLayer.java +++ b/deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/layers/objdetect/Yolo2OutputLayer.java @@ -191,8 +191,8 @@ public class Yolo2OutputLayer extends AbstractLayer