From e3a86292146a4e0f666c69495cd51ec93f808f9a Mon Sep 17 00:00:00 2001 From: Alex Black Date: Tue, 31 Mar 2020 11:38:12 +1100 Subject: [PATCH] 2 more try-with-resources (buffered + close) Signed-off-by: Alex Black --- .../java/org/deeplearning4j/util/ModelSerializer.java | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/util/ModelSerializer.java b/deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/util/ModelSerializer.java index 59820a082..997303977 100644 --- a/deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/util/ModelSerializer.java +++ b/deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/util/ModelSerializer.java @@ -415,7 +415,9 @@ public class ModelSerializer { */ public static Pair restoreMultiLayerNetworkAndNormalizer(@NonNull File file, boolean loadUpdater) throws IOException { - return restoreMultiLayerNetworkAndNormalizer(new FileInputStream(file), loadUpdater); + try(InputStream is = new BufferedInputStream(new FileInputStream(file))){ + return restoreMultiLayerNetworkAndNormalizer(is, loadUpdater); + } } /** @@ -876,8 +878,8 @@ public class ModelSerializer { * @return */ public static T restoreNormalizerFromFile(File file) throws IOException { - try { - return restoreNormalizerFromInputStream(new FileInputStream(file)); + try (InputStream is = new BufferedInputStream(new FileInputStream(file))) { + return restoreNormalizerFromInputStream(is); } catch (Exception e) { log.warn("Error while restoring normalizer, trying to restore assuming deprecated format..."); DataNormalization restoredDeprecated = restoreNormalizerFromInputStreamDeprecated(new FileInputStream(file)); @@ -918,8 +920,6 @@ public class ModelSerializer { * * This method restores normalizer from a given persisted model file serialized with Java object serialization * - * @param file - * @return */ private static DataNormalization restoreNormalizerFromInputStreamDeprecated(InputStream stream) { try {