Skip to content

Commit 68eba2e

Browse files
committed
try big endian
1 parent d170837 commit 68eba2e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/main/java/io/bioimage/modelrunner/pytorch/javacpp/shm/TensorBuilder.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ private static org.bytedeco.pytorch.Tensor buildFloat(SharedMemoryArray shmArray
135135
+ " is too big. Max number of elements per ubyte tensor supported: " + Integer.MAX_VALUE);
136136
if (!shmArray.isNumpyFormat())
137137
throw new IllegalArgumentException("Shared memory arrays must be saved in numpy format.");
138-
ByteBuffer buff = shmArray.getDataBufferNoHeader().order(ByteOrder.LITTLE_ENDIAN);
138+
ByteBuffer buff = shmArray.getDataBufferNoHeader().order(ByteOrder.BIG_ENDIAN);
139139
float[] flat = new float[buff.capacity() / 4];
140140
buff.asFloatBuffer().get(flat);
141141
Tensor ndarray = Tensor.create(flat, ogShape);

0 commit comments

Comments
 (0)