More large resource test tagging

master
agibsonccc 2021-03-26 10:13:27 +09:00
parent d2b133df87
commit 6e7a843d2d
6 changed files with 22 additions and 0 deletions

View File

@ -87,6 +87,8 @@ import static org.junit.jupiter.api.Assertions.*;
@Tag(TagNames.NDARRAY_ETL) @Tag(TagNames.NDARRAY_ETL)
@Tag(TagNames.TRAINING) @Tag(TagNames.TRAINING)
@Tag(TagNames.DL4J_OLD_API) @Tag(TagNames.DL4J_OLD_API)
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
public class TestEarlyStopping extends BaseDL4JTest { public class TestEarlyStopping extends BaseDL4JTest {

View File

@ -75,6 +75,8 @@ import static org.junit.jupiter.api.Assertions.*;
@Tag(TagNames.NDARRAY_ETL) @Tag(TagNames.NDARRAY_ETL)
@Tag(TagNames.TRAINING) @Tag(TagNames.TRAINING)
@Tag(TagNames.DL4J_OLD_API) @Tag(TagNames.DL4J_OLD_API)
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
public class TestEarlyStoppingCompGraph extends BaseDL4JTest { public class TestEarlyStoppingCompGraph extends BaseDL4JTest {
@Override @Override

View File

@ -155,6 +155,8 @@ public class RnnDataFormatTests extends BaseDL4JTest {
@MethodSource("org.deeplearning4j.nn.layers.recurrent.RnnDataFormatTests#params") @MethodSource("org.deeplearning4j.nn.layers.recurrent.RnnDataFormatTests#params")
@ParameterizedTest @ParameterizedTest
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
public void testGraveLSTM(boolean helpers, public void testGraveLSTM(boolean helpers,
boolean lastTimeStep, boolean lastTimeStep,
boolean maskZeros,Nd4jBackend backend) { boolean maskZeros,Nd4jBackend backend) {

View File

@ -38,12 +38,14 @@ import org.deeplearning4j.nn.gradient.Gradient;
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork; import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
import org.deeplearning4j.nn.workspace.ArrayType; import org.deeplearning4j.nn.workspace.ArrayType;
import org.deeplearning4j.nn.workspace.LayerWorkspaceMgr; import org.deeplearning4j.nn.workspace.LayerWorkspaceMgr;
import org.junit.jupiter.api.Tag;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedTest; import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments; import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource; import org.junit.jupiter.params.provider.MethodSource;
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.BaseNd4jTestWithBackends; import org.nd4j.linalg.BaseNd4jTestWithBackends;
import org.nd4j.linalg.activations.Activation; import org.nd4j.linalg.activations.Activation;
import org.nd4j.linalg.api.buffer.DataType; import org.nd4j.linalg.api.buffer.DataType;
@ -60,6 +62,8 @@ import java.util.stream.Stream;
import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertEquals;
@NativeTag @NativeTag
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
public class ConvDataFormatTests extends BaseDL4JTest { public class ConvDataFormatTests extends BaseDL4JTest {

View File

@ -44,7 +44,9 @@ import org.deeplearning4j.optimize.listeners.ScoreIterationListener;
import org.deeplearning4j.spark.earlystopping.SparkDataSetLossCalculator; import org.deeplearning4j.spark.earlystopping.SparkDataSetLossCalculator;
import org.deeplearning4j.spark.earlystopping.SparkEarlyStoppingTrainer; import org.deeplearning4j.spark.earlystopping.SparkEarlyStoppingTrainer;
import org.deeplearning4j.spark.impl.paramavg.ParameterAveragingTrainingMaster; import org.deeplearning4j.spark.impl.paramavg.ParameterAveragingTrainingMaster;
import org.junit.jupiter.api.Tag;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import org.nd4j.common.tests.tags.TagNames;
import org.nd4j.linalg.activations.Activation; import org.nd4j.linalg.activations.Activation;
import org.nd4j.linalg.dataset.DataSet; import org.nd4j.linalg.dataset.DataSet;
import org.nd4j.linalg.factory.Nd4j; import org.nd4j.linalg.factory.Nd4j;
@ -60,6 +62,10 @@ import java.util.concurrent.TimeUnit;
import static org.junit.jupiter.api.Assertions.*; import static org.junit.jupiter.api.Assertions.*;
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
@Tag(TagNames.DIST_SYSTEMS)
@Tag(TagNames.SPARK)
public class TestEarlyStoppingSpark extends BaseSparkTest { public class TestEarlyStoppingSpark extends BaseSparkTest {
@Test @Test

View File

@ -46,7 +46,9 @@ import org.deeplearning4j.spark.earlystopping.SparkEarlyStoppingGraphTrainer;
import org.deeplearning4j.spark.earlystopping.SparkLossCalculatorComputationGraph; import org.deeplearning4j.spark.earlystopping.SparkLossCalculatorComputationGraph;
import org.deeplearning4j.spark.impl.graph.dataset.DataSetToMultiDataSetFn; import org.deeplearning4j.spark.impl.graph.dataset.DataSetToMultiDataSetFn;
import org.deeplearning4j.spark.impl.paramavg.ParameterAveragingTrainingMaster; import org.deeplearning4j.spark.impl.paramavg.ParameterAveragingTrainingMaster;
import org.junit.jupiter.api.Tag;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import org.nd4j.common.tests.tags.TagNames;
import org.nd4j.linalg.activations.Activation; import org.nd4j.linalg.activations.Activation;
import org.nd4j.linalg.dataset.DataSet; import org.nd4j.linalg.dataset.DataSet;
import org.nd4j.linalg.factory.Nd4j; import org.nd4j.linalg.factory.Nd4j;
@ -62,6 +64,10 @@ import java.util.concurrent.TimeUnit;
import static org.junit.jupiter.api.Assertions.*; import static org.junit.jupiter.api.Assertions.*;
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
@Tag(TagNames.DIST_SYSTEMS)
@Tag(TagNames.SPARK)
public class TestEarlyStoppingSparkCompGraph extends BaseSparkTest { public class TestEarlyStoppingSparkCompGraph extends BaseSparkTest {