Skip to content

Commit 049b25d

Browse files
authored
Update preset map values for cspnet (#2416)
1 parent caf4890 commit 049b25d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tools/checkpoint_conversion/convert_cspnet_checkpoints.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,9 @@
2525

2626
PRESET_MAP = {
2727
"csp_darknet_53_ra_imagenet": "timm/cspdarknet53.ra_in1k",
28-
"csp_resnext_50_ra_imagenet": "cspresnext50.ra_in1k",
29-
"csp_resnet_50_ra_imagenet": "cspresnet50.ra_in1k",
30-
"darknet_53_imagenet": "darknet53.c2ns_in1k",
28+
"csp_resnext_50_ra_imagenet": "timm/cspresnext50.ra_in1k",
29+
"csp_resnet_50_ra_imagenet": "timm/cspresnet50.ra_in1k",
30+
"darknet_53_imagenet": "timm/darknet53.c2ns_in1k",
3131
}
3232
FLAGS = flags.FLAGS
3333

0 commit comments

Comments
 (0)