Remove additional batch size checks in training (#1129)

anwai98 · web-flow · commit d4c67c3995bf · 2025-12-17T19:58:25.000+01:00
diff --git a/micro_sam/training/sam_trainer.py b/micro_sam/training/sam_trainer.py
@@ -135,8 +135,6 @@ def _compute_loss(self, batched_outputs, y_one_hot):
         """
         mask_loss, iou_regression_loss = 0.0, 0.0
         batch_size = len(batched_outputs)
-        if batch_size == 0:
-            raise RuntimeError("Got empty batch outputs in loss computation.")
 
         # Loop over the batch.
         for batch_output, targets in zip(batched_outputs, y_one_hot):
