diff --git a/brutex-extended-tests/src/test/java/net/brutex/gan/App.java b/brutex-extended-tests/src/test/java/net/brutex/gan/App.java index dc5898413..6395223ad 100644 --- a/brutex-extended-tests/src/test/java/net/brutex/gan/App.java +++ b/brutex-extended-tests/src/test/java/net/brutex/gan/App.java @@ -16,6 +16,7 @@ import org.deeplearning4j.nn.conf.layers.misc.FrozenLayerWithBackprop; import org.deeplearning4j.nn.multilayer.MultiLayerNetwork; import org.deeplearning4j.nn.weights.WeightInit; import org.deeplearning4j.optimize.listeners.PerformanceListener; +import org.junit.jupiter.api.Tag; import org.junit.jupiter.api.Test; import org.nd4j.linalg.activations.Activation; import org.nd4j.linalg.activations.impl.ActivationLReLU; @@ -122,7 +123,7 @@ public class App { return conf; } - @Test + @Test @Tag("long-running") public void runTest() throws Exception { App.main(null); } diff --git a/brutex-extended-tests/src/test/java/net/brutex/gan/App2.java b/brutex-extended-tests/src/test/java/net/brutex/gan/App2.java index f9f7aba58..73e271fe1 100644 --- a/brutex-extended-tests/src/test/java/net/brutex/gan/App2.java +++ b/brutex-extended-tests/src/test/java/net/brutex/gan/App2.java @@ -44,6 +44,7 @@ import org.deeplearning4j.nn.conf.layers.*; import org.deeplearning4j.nn.conf.layers.misc.FrozenLayerWithBackprop; import org.deeplearning4j.nn.multilayer.MultiLayerNetwork; import org.deeplearning4j.optimize.listeners.PerformanceListener; +import org.junit.jupiter.api.Tag; import org.junit.jupiter.api.Test; import org.nd4j.linalg.api.ndarray.INDArray; import org.nd4j.linalg.dataset.DataSet; @@ -70,7 +71,7 @@ public class App2 { final static INDArray label_real = Nd4j.ones(BATCHSIZE, 1); final static INDArray label_fake = Nd4j.zeros(BATCHSIZE, 1); - @Test + @Test @Tag("long-running") void runTest() throws IOException { Nd4j.getMemoryManager().setAutoGcWindow(15 * 1000); diff --git a/brutex-extended-tests/src/test/java/net/brutex/gan/MnistSimpleGAN.java b/brutex-extended-tests/src/test/java/net/brutex/gan/MnistSimpleGAN.java index 2cfddb795..a31e9298f 100644 --- a/brutex-extended-tests/src/test/java/net/brutex/gan/MnistSimpleGAN.java +++ b/brutex-extended-tests/src/test/java/net/brutex/gan/MnistSimpleGAN.java @@ -31,6 +31,7 @@ import org.deeplearning4j.nn.conf.layers.DropoutLayer; import org.deeplearning4j.nn.conf.layers.OutputLayer; import org.deeplearning4j.nn.multilayer.MultiLayerNetwork; import org.deeplearning4j.nn.weights.WeightInit; +import org.junit.jupiter.api.Tag; import org.junit.jupiter.api.Test; import org.nd4j.linalg.activations.Activation; import org.nd4j.linalg.activations.impl.ActivationLReLU; @@ -100,7 +101,7 @@ public class MnistSimpleGAN { return new MultiLayerNetwork(discConf); } - @Test + @Test @Tag("long-running") public void runTest() throws Exception { main(null); }