♻️ [Update] load weight logging with min output

Author: henrytsui000

yolo/model/yolo.py

@@ -151,8 +151,14 @@ def save_load_weights(self, weights: Union[Path, OrderedDict]):
             model_state_dict[model_key] = weights[model_key]
 
         for error_name, error_set in error_dict.items():
-            for weight_name in error_set:
-                logger.warning(f":warning: Weight {error_name} for key: {'.'.join(weight_name)}")
+            error_dict = dict()
+            for layer_idx, *layer_name in error_set:
+                if layer_idx not in error_dict:
+                    error_dict[layer_idx] = [".".join(layer_name)]
+                else:
+                    error_dict[layer_idx].append(".".join(layer_name))
+            for layer_idx, layer_name in error_dict.items():
+                logger.warning(f":warning: Weight {error_name} for Layer {layer_idx}: {', '.join(layer_name)}")
 
         self.model.load_state_dict(model_state_dict)