🐞 Benchmark fixes for 2.5 (#4471)

Bug fixes

- Max epochs in train overrides the max_epochs value loaded from config when creating the engine
- Other fixes for benchmarking script

Signed-off-by: Ashwin Vaidya <[email protected]>

Author: ashwinvaidya17

src/otx/backend/native/engine.py

@@ -147,11 +147,11 @@ def __init__(
 
     def train(
         self,
-        max_epochs: int = 200,
+        max_epochs: int | None = None,
         min_epochs: int = 1,
         seed: int | None = None,
         deterministic: bool | Literal["warn"] = False,
-        precision: _PRECISION_INPUT | None = "16",
+        precision: _PRECISION_INPUT | None = None,
         val_check_interval: int | float | None = None,
         callbacks: list[Callback] | Callback | None = None,
         logger: Logger | Iterable[Logger] | bool | None = None,

src/otx/data/dataset/anomaly.py

@@ -80,7 +80,7 @@ def _get_item_impl(
                 ori_shape=img_shape,
                 image_color_channel=self.image_color_channel,
             ),
-            label=torch.tensor(label, dtype=torch.long),
+            label=label.to(dtype=torch.long),
             masks=Mask(self._get_mask(datumaro_item, label, img_shape)),
         )
 
@@ -159,7 +159,7 @@ def _mask_image_from_file(self, datumaro_item: DatasetItem, img_shape: tuple[int
         mask_file_path = (
             Path("/".join(datumaro_item.media.path.split("/")[:-3]))
             / "ground_truth"
-            / f"{('/'.join(datumaro_item.media.path.split('/')[-2:])).replace('.png','_mask.png')}"
+            / f"{('/'.join(datumaro_item.media.path.split('/')[-2:])).replace('.png', '_mask.png')}"
         )
         if mask_file_path.exists():
             return (io.read_image(str(mask_file_path), mode=io.ImageReadMode.GRAY) / 255).to(torch.uint8)