PLFM-3331 Register experiments with model name (#585)

* PLFM-3331 Register experiments with model name

* PLFM-3331 Fix black

requirements.pro.txt

@@ -1,2 +1,2 @@
-deci-lab-client==3.1.1
-deci-common==3.5.0
+deci-lab-client==4.8.0
+deci-common==3.15.0

src/super_gradients/common/sg_loggers/deci_platform_sg_logger.py

@@ -1,4 +1,5 @@
 import os
+from typing import Optional
 
 from super_gradients.common.abstractions.abstract_logger import get_logger
 from super_gradients.common.sg_loggers.base_sg_logger import BaseSGLogger
@@ -36,6 +37,7 @@ def __init__(
         save_tensorboard_remote: bool = True,
         save_logs_remote: bool = True,
         monitor_system: bool = True,
+        model_name: Optional[str] = None,
     ):
 
         if _imported_deci_lab_failure is not None:
@@ -59,7 +61,13 @@ def __init__(
 
         self.platform_client = DeciPlatformClient()
         self.platform_client.login(token=os.getenv("DECI_PLATFORM_TOKEN"))
-        self.platform_client.register_experiment(name=experiment_name)
+        if model_name is None:
+            logger.warning(
+                "'model_name' parameter not passed. "
+                "The experiment won't be connected to an architecture in the Deci platform. "
+                "To pass a model_name, please use the 'sg_logger_params.model_name' field in the training recipe."
+            )
+        self.platform_client.register_experiment(name=experiment_name, model_name=model_name if model_name else None)
         self.checkpoints_dir_path = checkpoints_dir_path
 
     @multi_process_safe