From 8a30ca1367923dc7539a8a45532d4aaaaaf23a17 Mon Sep 17 00:00:00 2001
From: agibsonccc <adam@konduit.ai>
Date: Fri, 26 Mar 2021 13:19:20 +0900
Subject: [PATCH] More test tagging

---
 .../deeplearning4j/nn/graph/TestComputationGraphNetwork.java    | 2 ++
 .../nn/layers/convolution/ConvolutionLayerTest.java             | 2 ++
 2 files changed, 4 insertions(+)

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