More large resources

master
agibsonccc 2021-03-26 12:16:51 +09:00
parent cd7b62dba8
commit f508c0e47b
3 changed files with 10 additions and 0 deletions

View File

@ -46,6 +46,8 @@ import java.util.concurrent.CountDownLatch;
public class RandomTests extends BaseDL4JTest { public class RandomTests extends BaseDL4JTest {
@Test @Test
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
public void testReproduce() throws Exception { public void testReproduce() throws Exception {
final MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder().updater(new RmsProp()) final MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder().updater(new RmsProp())

View File

@ -292,6 +292,8 @@ public class TestComputationGraphNetwork extends BaseDL4JTest {
} }
@Test @Test
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
public void testIrisFit() { public void testIrisFit() {
ComputationGraphConfiguration configuration = getIrisGraphConfiguration(); ComputationGraphConfiguration configuration = getIrisGraphConfiguration();
@ -1267,6 +1269,8 @@ public class TestComputationGraphNetwork extends BaseDL4JTest {
} }
@Test @Test
@Tag(TagNames.LONG_TEST)
@Tag(TagNames.LARGE_RESOURCES)
public void testEpochCounter() throws Exception { public void testEpochCounter() throws Exception {
ComputationGraphConfiguration conf = new NeuralNetConfiguration.Builder() ComputationGraphConfiguration conf = new NeuralNetConfiguration.Builder()
@ -1304,6 +1308,8 @@ public class TestComputationGraphNetwork extends BaseDL4JTest {
} }
@Test @Test
@Tag(TagNames.LONG_TEST)
@Tag(TagNames.LARGE_RESOURCES)
public void testSummary() { public void testSummary() {
int V_WIDTH = 130; int V_WIDTH = 130;
int V_HEIGHT = 130; int V_HEIGHT = 130;

View File

@ -502,6 +502,8 @@ class BatchNormalizationTest extends BaseDL4JTest {
@Test @Test
@DisplayName("Check Mean Variance Estimate CNN Compare Modes") @DisplayName("Check Mean Variance Estimate CNN Compare Modes")
@Tag(TagNames.LONG_TEST)
@Tag(TagNames.LARGE_RESOURCES)
void checkMeanVarianceEstimateCNNCompareModes() throws Exception { void checkMeanVarianceEstimateCNNCompareModes() throws Exception {
Nd4j.getRandom().setSeed(12345); Nd4j.getRandom().setSeed(12345);
// Check that the internal global mean/variance estimate is approximately correct // Check that the internal global mean/variance estimate is approximately correct