Merge branch 'dev' into feature/segmentation-ddrnet

luxonis_train/utils/config.py

@@ -82,28 +82,6 @@ class ModelConfig(BaseModelExtraForbid):
     visualizers: list[AttachedModuleConfig] = []
     outputs: list[str] = []
 
-    @model_validator(mode="after")
-    def check_main_metric(self) -> Self:
-        for metric in self.metrics:
-            if metric.is_main_metric:
-                logger.info(f"Main metric: `{metric.name}`")
-                return self
-
-        logger.warning("No main metric specified.")
-        if self.metrics:
-            metric = self.metrics[0]
-            metric.is_main_metric = True
-            name = metric.alias or metric.name
-            logger.info(f"Setting '{name}' as main metric.")
-        else:
-            logger.warning(
-                "[Ignore if using predefined model] "
-                "No metrics specified. "
-                "This is likely unintended unless "
-                "the configuration is not used for training. "
-            )
-        return self
-
     @model_validator(mode="after")
     def check_predefined_model(self) -> Self:
         from luxonis_train.utils.registry import MODELS
@@ -128,6 +106,28 @@ def check_predefined_model(self) -> Self:
 
         return self
 
+    @model_validator(mode="after")
+    def check_main_metric(self) -> Self:
+        for metric in self.metrics:
+            if metric.is_main_metric:
+                logger.info(f"Main metric: `{metric.name}`")
+                return self
+
+        logger.warning("No main metric specified.")
+        if self.metrics:
+            metric = self.metrics[0]
+            metric.is_main_metric = True
+            name = metric.alias or metric.name
+            logger.info(f"Setting '{name}' as main metric.")
+        else:
+            logger.warning(
+                "[Ignore if using predefined model] "
+                "No metrics specified. "
+                "This is likely unintended unless "
+                "the configuration is not used for training. "
+            )
+        return self
+
     @model_validator(mode="after")
     def check_graph(self) -> Self:
         from luxonis_train.utils import is_acyclic