diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java index ca0a4f5..7aacf76 100644 --- a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java +++ b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java @@ -514,16 +514,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