master
agibsonccc 2021-03-12 10:32:24 +09:00
parent d1734b4d25
commit 950db7ebe0
2 changed files with 4 additions and 0 deletions

View File

@ -24,6 +24,7 @@ import org.apache.commons.io.FileUtils;
import org.deeplearning4j.BaseDL4JTest; import org.deeplearning4j.BaseDL4JTest;
import org.deeplearning4j.nn.modelimport.keras.utils.DL4JKerasModelValidator; import org.deeplearning4j.nn.modelimport.keras.utils.DL4JKerasModelValidator;
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork; import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
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;
@ -41,6 +42,7 @@ import java.util.concurrent.atomic.AtomicInteger;
import static org.junit.Assert.*; import static org.junit.Assert.*;
@Ignore
public class MiscTests extends BaseDL4JTest { public class MiscTests extends BaseDL4JTest {
@Rule @Rule

View File

@ -31,6 +31,7 @@ import org.deeplearning4j.nn.modelimport.keras.KerasModelImport;
import org.deeplearning4j.nn.modelimport.keras.exceptions.InvalidKerasConfigurationException; import org.deeplearning4j.nn.modelimport.keras.exceptions.InvalidKerasConfigurationException;
import org.deeplearning4j.nn.modelimport.keras.exceptions.UnsupportedKerasConfigurationException; import org.deeplearning4j.nn.modelimport.keras.exceptions.UnsupportedKerasConfigurationException;
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork; import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
import org.junit.Ignore;
import org.junit.Test; import org.junit.Test;
import org.nd4j.common.resources.Resources; import org.nd4j.common.resources.Resources;
import org.nd4j.linalg.convolution.Convolution; import org.nd4j.linalg.convolution.Convolution;
@ -58,6 +59,7 @@ public class KerasModelImportTest extends BaseDL4JTest {
} }
@Test @Test
@Ignore
public void testNCHWNWHCChangeImport() { public void testNCHWNWHCChangeImport() {
MultiLayerNetwork model = loadModel("modelimport/keras/weights/conv2dnchw/simpleconv2d.hdf5"); MultiLayerNetwork model = loadModel("modelimport/keras/weights/conv2dnchw/simpleconv2d.hdf5");
MultiLayerConfiguration multiLayerConfiguration = model.getLayerWiseConfigurations(); MultiLayerConfiguration multiLayerConfiguration = model.getLayerWiseConfigurations();