handle crashes gracefully

Author: carlosuc3m

src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java

@@ -199,8 +199,11 @@ private void launchModelLoadOnProcess() throws IOException, InterruptedException
 			throw new RuntimeException();
 		else if (task.status == TaskStatus.FAILED)
 			throw new RuntimeException();
-		else if (task.status == TaskStatus.CRASHED)
+		else if (task.status == TaskStatus.CRASHED) {
+			this.runner.close();
+			runner = null;
 			throw new RuntimeException();
+		}
 	}
 
 	/**
@@ -321,8 +324,11 @@ void runInterprocessing(List<Tensor<T>> inputTensors, List<Tensor<R>> outputTens
 				throw new RuntimeException();
 			else if (task.status == TaskStatus.FAILED)
 				throw new RuntimeException();
-			else if (task.status == TaskStatus.CRASHED)
+			else if (task.status == TaskStatus.CRASHED) {
+				this.runner.close();
+				runner = null;
 				throw new RuntimeException();
+			}
 			for (int i = 0; i < outputTensors.size(); i ++) {
 	        	String name = (String) Types.decode(encOuts.get(i)).get(MEM_NAME_KEY);
 	        	SharedMemoryArray shm = shmaOutputList.stream()
@@ -455,8 +461,11 @@ public void closeModel() {
 				throw new RuntimeException();
 			else if (task.status == TaskStatus.FAILED)
 				throw new RuntimeException();
-			else if (task.status == TaskStatus.CRASHED)
+			else if (task.status == TaskStatus.CRASHED) {
+				this.runner.close();
+				runner = null;
 				throw new RuntimeException();
+			}
 			this.runner.close();
 			this.runner = null;
 			return;