try saveing more along the way

Author: carlosuc3m

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

@@ -275,6 +275,7 @@ protected void runFromShmas(List<String> inputs, List<String> outputs) throws IO
 			for (String ee : inputs) {
 				Map<String, Object> decoded = Types.decode(ee);
 				SharedMemoryArray shma = SharedMemoryArray.read((String) decoded.get(MEM_NAME_KEY));
+	        	DecodeNumpy.saveNpy("/home/carlos/git/mm_in.npy", Cast.unchecked(shma.getSharedRAI()));
 				NDArray inT = TensorBuilder.build(shma, manager);
 				if (PlatformDetection.isWindows()) shma.close();
 				inputList.add(inT);
@@ -339,7 +340,6 @@ 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) {

src/main/java/io/bioimage/modelrunner/pytorch/shm/ShmBuilder.java

@@ -20,6 +20,7 @@
  */
 package io.bioimage.modelrunner.pytorch.shm;
 
+import io.bioimage.modelrunner.numpy.DecodeNumpy;
 import io.bioimage.modelrunner.system.PlatformDetection;
 import io.bioimage.modelrunner.tensor.shm.SharedMemoryArray;
 import io.bioimage.modelrunner.utils.CommonUtils;
@@ -33,6 +34,7 @@
 import net.imglib2.type.numeric.integer.UnsignedByteType;
 import net.imglib2.type.numeric.real.DoubleType;
 import net.imglib2.type.numeric.real.FloatType;
+import net.imglib2.util.Cast;
 
 /**
  * A utility class that converts {@link NDArray}s into {@link SharedMemoryArray}s for
@@ -114,6 +116,7 @@ private static void buildFromTensorFloat(NDArray tensor, String memoryName) thro
 
         SharedMemoryArray shma = SharedMemoryArray.readOrCreate(memoryName, arrayShape, new FloatType(), false, true);
         shma.getDataBufferNoHeader().put(tensor.toByteArray());
+    	DecodeNumpy.saveNpy("/home/carlos/git/mm_out.npy", Cast.unchecked(shma.getSharedRAI()));
         if (PlatformDetection.isWindows()) shma.close();
     }