line for debugging

bioimage-io · Dec 20, 2024 · 1d90073 · 1d90073
1 parent 552869e
commit 1d90073
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java b/src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java
@@ -27,6 +27,7 @@
 import io.bioimage.modelrunner.engine.DeepLearningEngineInterface;
 import io.bioimage.modelrunner.exceptions.LoadModelException;
 import io.bioimage.modelrunner.exceptions.RunModelException;
+import io.bioimage.modelrunner.numpy.DecodeNumpy;
 import io.bioimage.modelrunner.pytorch.shm.ShmBuilder;
 import io.bioimage.modelrunner.pytorch.shm.TensorBuilder;
 import io.bioimage.modelrunner.pytorch.tensor.ImgLib2Builder;
@@ -338,6 +339,7 @@ else if (task.status == TaskStatus.CRASHED) {
 	        		shmaOutputList.add(shm);
 	        	}
 	        	RandomAccessibleInterval<?> rai = shm.getSharedRAI();
+	        	DecodeNumpy.saveNpy("/home/carlos/git/mm.npy", Cast.unchecked(rai));
 	        	outputTensors.get(i).setData(Tensor.createCopyOfRaiInWantedDataType(Cast.unchecked(rai), Util.getTypeFromInterval(Cast.unchecked(rai))));
 	        }
 		} catch (Exception e) {