diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/RandomTests.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/RandomTests.java index f4bae120e..01df1b517 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/RandomTests.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/RandomTests.java @@ -46,6 +46,8 @@ import java.util.concurrent.CountDownLatch; public class RandomTests extends BaseDL4JTest { @Test + @Tag(TagNames.LARGE_RESOURCES) + @Tag(TagNames.LONG_TEST) public void testReproduce() throws Exception { final MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder().updater(new RmsProp()) diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/graph/TestComputationGraphNetwork.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/graph/TestComputationGraphNetwork.java index ed16e0abf..7095de38d 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/graph/TestComputationGraphNetwork.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/graph/TestComputationGraphNetwork.java @@ -292,6 +292,8 @@ public class TestComputationGraphNetwork extends BaseDL4JTest { } @Test + @Tag(TagNames.LARGE_RESOURCES) + @Tag(TagNames.LONG_TEST) public void testIrisFit() { ComputationGraphConfiguration configuration = getIrisGraphConfiguration(); @@ -1267,6 +1269,8 @@ public class TestComputationGraphNetwork extends BaseDL4JTest { } @Test + @Tag(TagNames.LONG_TEST) + @Tag(TagNames.LARGE_RESOURCES) public void testEpochCounter() throws Exception { ComputationGraphConfiguration conf = new NeuralNetConfiguration.Builder() @@ -1304,6 +1308,8 @@ public class TestComputationGraphNetwork extends BaseDL4JTest { } @Test + @Tag(TagNames.LONG_TEST) + @Tag(TagNames.LARGE_RESOURCES) public void testSummary() { int V_WIDTH = 130; int V_HEIGHT = 130; 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 09d2ffb82..cb84aecee 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 @@ -502,6 +502,8 @@ class BatchNormalizationTest extends BaseDL4JTest { @Test @DisplayName("Check Mean Variance Estimate CNN Compare Modes") + @Tag(TagNames.LONG_TEST) + @Tag(TagNames.LARGE_RESOURCES) void checkMeanVarianceEstimateCNNCompareModes() throws Exception { Nd4j.getRandom().setSeed(12345); // Check that the internal global mean/variance estimate is approximately correct