fix batch size to 1 to avoid onnx error

bioimage-io · Jul 17, 2024 · c092b8b · c092b8b
1 parent fbf2415
commit c092b8b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/example_descriptions/models/unet2d_nuclei_broad/bioimageio.yaml b/example_descriptions/models/unet2d_nuclei_broad/bioimageio.yaml
@@ -37,6 +37,7 @@ inputs:
     description: raw input
     axes:
       - type: batch
+        size: 1 # due to the onnx weights
       - type: channel
         channel_names: [raw_intensity]
       - type: space # todo: scale/unit