update

CodSpeedHQ · Sep 4, 2024 · dcf3768 · dcf3768
1 parent 8764caa
commit dcf3768
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kornia/contrib/models/rt_detr/model.py b/kornia/contrib/models/rt_detr/model.py
@@ -180,9 +180,9 @@ def from_pretrained(model_name: str) -> RTDETR:
         )
 
         def map_name(old_name: str) -> str:
-            # Start with the old name
             new_name = old_name
 
+            # Encoder renaming
             new_name = re.sub("encoder.pan_blocks", "encoder.ccfm.pan_blocks", new_name)
             new_name = re.sub("encoder.downsample_convs", "encoder.ccfm.downsample_convs", new_name)
             new_name = re.sub("encoder.fpn_blocks", "encoder.ccfm.fpn_blocks", new_name)