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 7095de38d..7d452db56 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 @@ -599,6 +599,8 @@ public class TestComputationGraphNetwork extends BaseDL4JTest { } @Test + @Tag(TagNames.LONG_TEST) + @Tag(TagNames.LARGE_RESOURCES) public void testPreTraining() { ComputationGraphConfiguration conf = new NeuralNetConfiguration.Builder() diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/convolution/ConvolutionLayerTest.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/convolution/ConvolutionLayerTest.java index 1f659fa06..854fbcf92 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/convolution/ConvolutionLayerTest.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/convolution/ConvolutionLayerTest.java @@ -74,6 +74,8 @@ import static org.junit.jupiter.api.Assertions.assertThrows; @DisplayName("Convolution Layer Test") @NativeTag @Tag(TagNames.DL4J_OLD_API) +@Tag(TagNames.LARGE_RESOURCES) +@Tag(TagNames.LONG_TEST) class ConvolutionLayerTest extends BaseDL4JTest { @Override