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

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