Changed model names to reflect the actual models

configs/classification_heavy_model.yaml

@@ -1,7 +1,7 @@
 # Example configuration for training a predefined heavy classification model
 
 model:
-  name: cifar10_classification
+  name: segmentation_light
   predefined_model:
     name: ClassificationModel
     params:

configs/classification_light_model.yaml

@@ -1,7 +1,7 @@
 # Example configuration for training a predefined light classification model
 
 model:
-  name: cifar10_classification
+  name: segmentation_light
   predefined_model:
     name: ClassificationModel
     params:

configs/detection_heavy_model.yaml

@@ -1,7 +1,7 @@
 # Example configuration for training a predefined heavy detection model
 
 model:
-  name: coco_detection
+  name: detection_heavy
   predefined_model:
     name: DetectionModel
     params:

configs/detection_light_model.yaml

@@ -1,7 +1,7 @@
 # Example configuration for training a predefined light detection model
 
 model:
-  name: coco_detection
+  name: detection_light
   predefined_model:
     name: DetectionModel
     params:

configs/keypoint_bbox_heavy_model.yaml

@@ -1,7 +1,7 @@
 # Example configuration for training a predefined heavy keypoint-detection model
 
 model:
-  name: coco_keypoints
+  name: keypoint_detection_heavy
   predefined_model:
     name: KeypointDetectionModel
     params:

configs/keypoint_bbox_light_model.yaml

@@ -1,7 +1,7 @@
 # Example configuration for training a predefined light keypoint-detection model
 
 model:
-  name: coco_keypoints
+  name: keypoint_detection_light
   predefined_model:
     name: KeypointDetectionModel
     params:

configs/segmentation_heavy_model.yaml

@@ -1,8 +1,7 @@
-# DDRNet-23-slim model for segmentation
-# Refer to here for optimal hyperparameters for this model: https://github.com/Deci-AI/super-gradients/blob/4797c974c7c445d12e2575c468848d9c3e04becd/src/super_gradients/recipes/cityscapes_ddrnet.yaml#L4
+# Example configuration for training a predefined heavy segmentation model
 
 model:
-  name: ddrnet_segmentation
+  name: segmentation_heavy
   predefined_model:
     name: SegmentationModel
     params:

configs/segmentation_light_model.yaml

@@ -1,7 +1,7 @@
 # Example configuration for training a predefined light segmentation model
 
 model:
-  name: ddrnet_segmentation
+  name: segmentation_light
   predefined_model:
     name: SegmentationModel
     params: