From 0161038112151656270c78e43ac321daab0c016f Mon Sep 17 00:00:00 2001 From: agibsonccc Date: Fri, 26 Mar 2021 12:48:05 +0900 Subject: [PATCH] More test tagging --- .../org/deeplearning4j/nn/graph/TestCompGraphUnsupervised.java | 2 ++ .../nn/layers/convolution/ConvolutionLayerSetupTest.java | 1 + .../nn/layers/normalization/BatchNormalizationTest.java | 2 ++ .../java/org/deeplearning4j/nn/multilayer/MultiLayerTest.java | 3 +++ 4 files changed, 8 insertions(+) diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/graph/TestCompGraphUnsupervised.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/graph/TestCompGraphUnsupervised.java index 51cb46f5c..b947c0d86 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/graph/TestCompGraphUnsupervised.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/graph/TestCompGraphUnsupervised.java @@ -54,6 +54,8 @@ import static org.junit.jupiter.api.Assertions.assertNotEquals; @NativeTag @Tag(TagNames.DL4J_OLD_API) +@Tag(TagNames.LONG_TEST) +@Tag(TagNames.LARGE_RESOURCES) public class TestCompGraphUnsupervised extends BaseDL4JTest { @Override diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/convolution/ConvolutionLayerSetupTest.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/convolution/ConvolutionLayerSetupTest.java index 0db84bfc3..0b586f900 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/convolution/ConvolutionLayerSetupTest.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/convolution/ConvolutionLayerSetupTest.java @@ -65,6 +65,7 @@ import org.junit.jupiter.api.extension.ExtendWith; @DisplayName("Convolution Layer Setup Test") @NativeTag @Tag(TagNames.DL4J_OLD_API) +@Tag(TagNames.LARGE_RESOURCES) class ConvolutionLayerSetupTest extends BaseDL4JTest { @TempDir diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/normalization/BatchNormalizationTest.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/normalization/BatchNormalizationTest.java index cb84aecee..3e81dac9e 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/normalization/BatchNormalizationTest.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/normalization/BatchNormalizationTest.java @@ -80,6 +80,8 @@ import org.junit.jupiter.api.extension.ExtendWith; @DisplayName("Batch Normalization Test") @NativeTag @Tag(TagNames.DL4J_OLD_API) +@Tag(TagNames.LONG_TEST) +@Tag(TagNames.LARGE_RESOURCES) class BatchNormalizationTest extends BaseDL4JTest { static { diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/multilayer/MultiLayerTest.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/multilayer/MultiLayerTest.java index 16af69f02..80661a17c 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/multilayer/MultiLayerTest.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/multilayer/MultiLayerTest.java @@ -90,6 +90,8 @@ import static org.junit.jupiter.api.Assertions.assertThrows; @DisplayName("Multi Layer Test") @NativeTag @Tag(TagNames.DL4J_OLD_API) +@Tag(TagNames.LARGE_RESOURCES) +@Tag(TagNames.LONG_TEST) public class MultiLayerTest extends BaseDL4JTest { private static OpExecutioner.ProfilingMode origMode; @@ -872,6 +874,7 @@ public class MultiLayerTest extends BaseDL4JTest { @Test @DisplayName("Test Multiple Epochs Simple") + void testMultipleEpochsSimple() { // Mainly a simple sanity check on the preconditions in the method... DataSetIterator iter = new IrisDataSetIterator(10, 150);