From d2b133df87a02613a70a8e1485952817e033e81f Mon Sep 17 00:00:00 2001 From: agibsonccc Date: Fri, 26 Mar 2021 09:48:43 +0900 Subject: [PATCH] Update Keras2ModelConfigurationTest.java --- .../keras/configurations/Keras2ModelConfigurationTest.java | 2 ++ 1 file changed, 2 insertions(+) diff --git a/deeplearning4j/deeplearning4j-modelimport/src/test/java/org/deeplearning4j/nn/modelimport/keras/configurations/Keras2ModelConfigurationTest.java b/deeplearning4j/deeplearning4j-modelimport/src/test/java/org/deeplearning4j/nn/modelimport/keras/configurations/Keras2ModelConfigurationTest.java index d0749f9ae..e8243a0b4 100644 --- a/deeplearning4j/deeplearning4j-modelimport/src/test/java/org/deeplearning4j/nn/modelimport/keras/configurations/Keras2ModelConfigurationTest.java +++ b/deeplearning4j/deeplearning4j-modelimport/src/test/java/org/deeplearning4j/nn/modelimport/keras/configurations/Keras2ModelConfigurationTest.java @@ -293,6 +293,8 @@ class Keras2ModelConfigurationTest extends BaseDL4JTest { @Test @DisplayName("One Lstm Layer Test") + @Tag(TagNames.LARGE_RESOURCES) + @Tag(TagNames.LONG_TEST) void oneLstmLayerTest() throws Exception { 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();