soy tontoooooooooooo

bioimage-io · Nov 23, 2024 · e6f5bd5 · e6f5bd5
1 parent 58bd149
commit e6f5bd5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/shm/ShmBuilder.java b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/shm/ShmBuilder.java
@@ -117,7 +117,7 @@ private static void buildFromTensorFloat(Tensor tensor, String memoryName) throw
     	ByteBuffer byteBuffer = ByteBuffer.allocateDirect((int) (flatSize * Float.BYTES));
     	FloatBuffer floatBuffer = byteBuffer.asFloatBuffer();
     	tensor.data_ptr_float().get(flat);
-    	floatBuffer.put(flatSize);
+    	floatBuffer.put(flat);
     	byteBuffer.rewind();
         shma.getDataBufferNoHeader().put(byteBuffer);
         if (PlatformDetection.isWindows()) shma.close();