Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Keras2ModelConfigurationTest.java

master
agibsonccc 2021-03-26 09:48:43 +09:00
parent b3805a132f
commit d2b133df87
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deeplearning4j/deeplearning4j-modelimport/src/test/java/org/deeplearning4j/nn/modelimport/keras/configurations/Keras2ModelConfigurationTest.java
View File

@ -293,6 +293,8 @@ class Keras2ModelConfigurationTest extends BaseDL4JTest {
@Test @Test
@DisplayName("One Lstm Layer Test") @DisplayName("One Lstm Layer Test")
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
void oneLstmLayerTest() throws Exception { void oneLstmLayerTest() throws Exception {
try (InputStream is = Resources.asStream("/modelimport/keras/configs/keras2/one_lstm_no_sequences_tf_keras_2.json")) { try (InputStream is = Resources.asStream("/modelimport/keras/configs/keras2/one_lstm_no_sequences_tf_keras_2.json")) {
MultiLayerConfiguration config = new KerasModel().modelBuilder().modelJsonInputStream(is).enforceTrainingConfig(false).buildSequential().getMultiLayerConfiguration(); MultiLayerConfiguration config = new KerasModel().modelBuilder().modelJsonInputStream(is).enforceTrainingConfig(false).buildSequential().getMultiLayerConfiguration();