Skip to content

Commit 1d90073

Browse files
committed
line for debugging
1 parent 552869e commit 1d90073

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

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

+2
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import io.bioimage.modelrunner.engine.DeepLearningEngineInterface;
2828
import io.bioimage.modelrunner.exceptions.LoadModelException;
2929
import io.bioimage.modelrunner.exceptions.RunModelException;
30+
import io.bioimage.modelrunner.numpy.DecodeNumpy;
3031
import io.bioimage.modelrunner.pytorch.shm.ShmBuilder;
3132
import io.bioimage.modelrunner.pytorch.shm.TensorBuilder;
3233
import io.bioimage.modelrunner.pytorch.tensor.ImgLib2Builder;
@@ -338,6 +339,7 @@ else if (task.status == TaskStatus.CRASHED) {
338339
shmaOutputList.add(shm);
339340
}
340341
RandomAccessibleInterval<?> rai = shm.getSharedRAI();
342+
DecodeNumpy.saveNpy("/home/carlos/git/mm.npy", Cast.unchecked(rai));
341343
outputTensors.get(i).setData(Tensor.createCopyOfRaiInWantedDataType(Cast.unchecked(rai), Util.getTypeFromInterval(Cast.unchecked(rai))));
342344
}
343345
} catch (Exception e) {

0 commit comments

Comments
 (0)