Skip to content

Commit 16ba3d1

Browse files
committed
🎨 [Update] the mismatch log with sorted name
1 parent 50b6d7f commit 16ba3d1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

yolo/model/yolo.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,7 @@ def save_load_weights(self, weights: Union[Path, OrderedDict]):
158158
else:
159159
error_dict[layer_idx].append(".".join(layer_name))
160160
for layer_idx, layer_name in error_dict.items():
161+
layer_name.sort()
161162
logger.warning(f":warning: Weight {error_name} for Layer {layer_idx}: {', '.join(layer_name)}")
162163

163164
self.model.load_state_dict(model_state_dict)

0 commit comments

Comments
 (0)