diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationDataSetIteratorTest.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationDataSetIteratorTest.java index 65fff53a3..c0bd12c5b 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationDataSetIteratorTest.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationDataSetIteratorTest.java @@ -22,8 +22,10 @@ package org.deeplearning4j.datasets.iterator; import org.deeplearning4j.BaseDL4JTest; import org.deeplearning4j.datasets.iterator.impl.MnistDataSetIterator; +import org.junit.jupiter.api.Tag; import org.junit.jupiter.api.Test; import org.nd4j.common.tests.tags.NativeTag; +import org.nd4j.common.tests.tags.TagNames; import org.nd4j.linalg.dataset.DataSet; import org.nd4j.linalg.dataset.api.iterator.DataSetIterator; @@ -37,6 +39,8 @@ import org.junit.jupiter.api.DisplayName; @DisplayName("Early Termination Data Set Iterator Test") @NativeTag +@Tag(TagNames.LARGE_RESOURCES) +@Tag(TagNames.LONG_TEST) class EarlyTerminationDataSetIteratorTest extends BaseDL4JTest { int minibatchSize = 10; diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationMultiDataSetIteratorTest.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationMultiDataSetIteratorTest.java index 8413f57f7..a729d0d04 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationMultiDataSetIteratorTest.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationMultiDataSetIteratorTest.java @@ -40,6 +40,8 @@ import static org.junit.jupiter.api.Assertions.*; @DisplayName("Early Termination Multi Data Set Iterator Test") @NativeTag @Tag(TagNames.FILE_IO) +@Tag(TagNames.LARGE_RESOURCES) +@Tag(TagNames.LONG_TEST) class EarlyTerminationMultiDataSetIteratorTest extends BaseDL4JTest { int minibatchSize = 5;