More test tagging
parent
f508c0e47b
commit
0161038112
|
@ -54,6 +54,8 @@ import static org.junit.jupiter.api.Assertions.assertNotEquals;
|
||||||
|
|
||||||
@NativeTag
|
@NativeTag
|
||||||
@Tag(TagNames.DL4J_OLD_API)
|
@Tag(TagNames.DL4J_OLD_API)
|
||||||
|
@Tag(TagNames.LONG_TEST)
|
||||||
|
@Tag(TagNames.LARGE_RESOURCES)
|
||||||
public class TestCompGraphUnsupervised extends BaseDL4JTest {
|
public class TestCompGraphUnsupervised extends BaseDL4JTest {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
@ -65,6 +65,7 @@ import org.junit.jupiter.api.extension.ExtendWith;
|
||||||
@DisplayName("Convolution Layer Setup Test")
|
@DisplayName("Convolution Layer Setup Test")
|
||||||
@NativeTag
|
@NativeTag
|
||||||
@Tag(TagNames.DL4J_OLD_API)
|
@Tag(TagNames.DL4J_OLD_API)
|
||||||
|
@Tag(TagNames.LARGE_RESOURCES)
|
||||||
class ConvolutionLayerSetupTest extends BaseDL4JTest {
|
class ConvolutionLayerSetupTest extends BaseDL4JTest {
|
||||||
|
|
||||||
@TempDir
|
@TempDir
|
||||||
|
|
|
@ -80,6 +80,8 @@ import org.junit.jupiter.api.extension.ExtendWith;
|
||||||
@DisplayName("Batch Normalization Test")
|
@DisplayName("Batch Normalization Test")
|
||||||
@NativeTag
|
@NativeTag
|
||||||
@Tag(TagNames.DL4J_OLD_API)
|
@Tag(TagNames.DL4J_OLD_API)
|
||||||
|
@Tag(TagNames.LONG_TEST)
|
||||||
|
@Tag(TagNames.LARGE_RESOURCES)
|
||||||
class BatchNormalizationTest extends BaseDL4JTest {
|
class BatchNormalizationTest extends BaseDL4JTest {
|
||||||
|
|
||||||
static {
|
static {
|
||||||
|
|
|
@ -90,6 +90,8 @@ import static org.junit.jupiter.api.Assertions.assertThrows;
|
||||||
@DisplayName("Multi Layer Test")
|
@DisplayName("Multi Layer Test")
|
||||||
@NativeTag
|
@NativeTag
|
||||||
@Tag(TagNames.DL4J_OLD_API)
|
@Tag(TagNames.DL4J_OLD_API)
|
||||||
|
@Tag(TagNames.LARGE_RESOURCES)
|
||||||
|
@Tag(TagNames.LONG_TEST)
|
||||||
public class MultiLayerTest extends BaseDL4JTest {
|
public class MultiLayerTest extends BaseDL4JTest {
|
||||||
|
|
||||||
private static OpExecutioner.ProfilingMode origMode;
|
private static OpExecutioner.ProfilingMode origMode;
|
||||||
|
@ -872,6 +874,7 @@ public class MultiLayerTest extends BaseDL4JTest {
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
@DisplayName("Test Multiple Epochs Simple")
|
@DisplayName("Test Multiple Epochs Simple")
|
||||||
|
|
||||||
void testMultipleEpochsSimple() {
|
void testMultipleEpochsSimple() {
|
||||||
// Mainly a simple sanity check on the preconditions in the method...
|
// Mainly a simple sanity check on the preconditions in the method...
|
||||||
DataSetIterator iter = new IrisDataSetIterator(10, 150);
|
DataSetIterator iter = new IrisDataSetIterator(10, 150);
|
||||||
|
|
Loading…
Reference in New Issue