diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java index bb56000..63ba636 100644 --- a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java +++ b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java @@ -502,16 +502,17 @@ else if (task.status == TaskStatus.FAILED) else if (task.status == TaskStatus.CRASHED) throw new RuntimeException(); this.runner.close(); + this.runner = null; return; } else if (this.interprocessing) { return; } - sig = null; if (model != null) { model.session().close(); model.close(); } model = null; + sig = null; } // TODO make only one