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

API fix + try with resources (close file stream + buffered)

Browse Source 
Signed-off-by: Alex Black <blacka101@gmail.com>
This commit is contained in:
Alex Black 2020-03-31 11:34:43 +11:00
parent 2e52f0217f
commit 3f49d646fc
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deeplearning4j/deeplearning4j-core/src/main/java/org/deeplearning4j/util/ModelGuesser.java
View File

@ -54,7 +54,11 @@ public class ModelGuesser {
* @return the loaded normalizer
*/
public static Normalizer<?> loadNormalizer(String path) {
return ModelSerializer.restoreNormalizerFromFile(new File(path));
try {
return ModelSerializer.restoreNormalizerFromFile(new File(path));
} catch (IOException e){
throw new RuntimeException(e);
}
}

4
deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/util/ModelSerializer.java
View File

@ -218,7 +218,9 @@ public class ModelSerializer {
*/
public static MultiLayerNetwork restoreMultiLayerNetwork(@NonNull File file, boolean loadUpdater)
throws IOException {
return restoreMultiLayerNetwork(new FileInputStream(file), loadUpdater);
try(InputStream is = new BufferedInputStream(new FileInputStream(file))){
return restoreMultiLayerNetwork(is, loadUpdater);
}
}