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