Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

More test tagging

master
agibsonccc 2021-03-26 13:42:46 +09:00
parent 8a30ca1367
commit cf26ffe3d3
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationDataSetIteratorTest.java
View File

@ -22,8 +22,10 @@ package org.deeplearning4j.datasets.iterator;
import org.deeplearning4j.BaseDL4JTest; import org.deeplearning4j.BaseDL4JTest;
import org.deeplearning4j.datasets.iterator.impl.MnistDataSetIterator; import org.deeplearning4j.datasets.iterator.impl.MnistDataSetIterator;
import org.junit.jupiter.api.Tag;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import org.nd4j.common.tests.tags.NativeTag; 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.DataSet;
import org.nd4j.linalg.dataset.api.iterator.DataSetIterator; 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") @DisplayName("Early Termination Data Set Iterator Test")
@NativeTag @NativeTag
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
class EarlyTerminationDataSetIteratorTest extends BaseDL4JTest { class EarlyTerminationDataSetIteratorTest extends BaseDL4JTest {
int minibatchSize = 10; int minibatchSize = 10;

2
deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationMultiDataSetIteratorTest.java
View File

@ -40,6 +40,8 @@ import static org.junit.jupiter.api.Assertions.*;
@DisplayName("Early Termination Multi Data Set Iterator Test") @DisplayName("Early Termination Multi Data Set Iterator Test")
@NativeTag @NativeTag
@Tag(TagNames.FILE_IO) @Tag(TagNames.FILE_IO)
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
class EarlyTerminationMultiDataSetIteratorTest extends BaseDL4JTest { class EarlyTerminationMultiDataSetIteratorTest extends BaseDL4JTest {
int minibatchSize = 5; int minibatchSize = 5;