Signed-off-by: Alex Black <blacka101@gmail.com>master
parent
226f0672bc
commit
838c3ddb5a
|
@ -37,6 +37,11 @@ import static org.junit.Assert.assertEquals;
|
||||||
@Slf4j
|
@Slf4j
|
||||||
public class AssertTestsExtendBaseClass extends BaseDL4JTest {
|
public class AssertTestsExtendBaseClass extends BaseDL4JTest {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long getTimeoutMilliseconds() {
|
||||||
|
return 240000L;
|
||||||
|
}
|
||||||
|
|
||||||
//Set of classes that are exclusions to the rule (either run manually or have their own logging + timeouts)
|
//Set of classes that are exclusions to the rule (either run manually or have their own logging + timeouts)
|
||||||
private static final Set<Class<?>> exclusions = new HashSet<>();
|
private static final Set<Class<?>> exclusions = new HashSet<>();
|
||||||
|
|
||||||
|
|
|
@ -42,6 +42,11 @@ import static org.junit.Assert.assertEquals;
|
||||||
@Slf4j
|
@Slf4j
|
||||||
public class AssertTestsExtendBaseClass extends BaseND4JTest {
|
public class AssertTestsExtendBaseClass extends BaseND4JTest {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long getTimeoutMilliseconds() {
|
||||||
|
return 240000L;
|
||||||
|
}
|
||||||
|
|
||||||
//Set of classes that are exclusions to the rule (either run manually or have their own logging + timeouts)
|
//Set of classes that are exclusions to the rule (either run manually or have their own logging + timeouts)
|
||||||
private static final Set<Class<?>> exclusions = new HashSet<>(Arrays.asList(
|
private static final Set<Class<?>> exclusions = new HashSet<>(Arrays.asList(
|
||||||
TFGraphTestAllSameDiff.class,
|
TFGraphTestAllSameDiff.class,
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
package org.nd4j.autodiff.samediff;
|
package org.nd4j.autodiff.samediff;
|
||||||
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.junit.Ignore;
|
||||||
import org.junit.Rule;
|
import org.junit.Rule;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import org.junit.rules.TemporaryFolder;
|
import org.junit.rules.TemporaryFolder;
|
||||||
|
@ -82,6 +83,7 @@ public class SameDiffMultiThreadTests extends BaseND4JTest {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
|
@Ignore //2020/03/24 AB - https://github.com/eclipse/deeplearning4j/issues/8802
|
||||||
public void testMobilenet() throws Exception {
|
public void testMobilenet() throws Exception {
|
||||||
TFGraphTestZooModels.currentTestDir = testDir.newFolder();
|
TFGraphTestZooModels.currentTestDir = testDir.newFolder();
|
||||||
File f = Resources.asFile("tf_graphs/zoo_models/mobilenet_v2_1.0_224/tf_model.txt");
|
File f = Resources.asFile("tf_graphs/zoo_models/mobilenet_v2_1.0_224/tf_model.txt");
|
||||||
|
|
Loading…
Reference in New Issue