Fix loading Yolov8 sparsezoo models (#1561) (#1565)

* Fix loading Yolov8 sparsezoo models

* quality checks

Author: abhinavnmagic

src/sparseml/yolov8/trainers.py

@@ -533,6 +533,7 @@ def __init__(self, model="yolov8n.yaml", type="v8") -> None:
             model = download_framework_model_by_recipe_type(
                 Model(model_str), model_suffix="pt"
             )
+            model_str = str(model)
             self.is_sparseml_checkpoint = True
 
         if model_str.endswith(".pt"):
@@ -793,6 +794,7 @@ def val(self, data=None, **kwargs):
         overrides["rect"] = True  # rect batches as default
         overrides.update(kwargs)
         overrides["mode"] = "val"
+        overrides["data"] = data or overrides["data"]
         args = get_cfg(cfg=DEFAULT_CFG, overrides=overrides)
         args.data = data or args.data
         args.task = self.task

src/sparseml/yolov8/val.py

@@ -76,6 +76,7 @@
     default=None,
     help="Path to override default datasets path.",
 )
+@click.option("--batch", default=16, type=int, help="number of images per batch")
 def main(**kwargs):
     if kwargs["dataset_path"] is not None:
         kwargs["data"] = data_from_dataset_path(kwargs["data"], kwargs["dataset_path"])