Skip to content
Snippets Groups Projects
Commit f7179368 authored by carlosuc3m's avatar carlosuc3m
Browse files

close model if not null

parent e4b46197
Branches
No related tags found
No related merge requests found
......@@ -132,9 +132,10 @@ public class Tensorflow1Interface implements DeepLearningInterface
@Override
public void closeModel() {
sig = null;
model.session().close();
if (model != null)
if (model != null) {
model.session().close();
model.close();
}
model = null;
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment