More test tagging

master
agibsonccc 2021-03-26 12:48:05 +09:00
parent f508c0e47b
commit 0161038112
4 changed files with 8 additions and 0 deletions

View File

@ -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

View File

@ -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

View File

@ -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 {

View File

@ -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);