properly close model

bioimage-io · Oct 3, 2024 · e21a11f · e21a11f
1 parent b4f0158
commit e21a11f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/PytorchJavaCPPInterface.java b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/PytorchJavaCPPInterface.java
@@ -367,6 +367,7 @@ 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;
@@ -375,6 +376,7 @@ else if (task.status == TaskStatus.CRASHED)
 			return;
 		model.close();
 		model.deallocate();
+		model = null;
 	}
 
 	/**