Skip to content

Commit 9bfc101

Browse files
committed
feat: update pretrained weights descriptions for ResNet models
- Refined descriptions for ResNet-18, ResNet-32, and ResNet-50 pretrained weights. - Ensured consistent naming aligned with origin frameworks (TorchVision, RSB Paper).
1 parent 297337a commit 9bfc101

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

crates/bimm/src/models/resnet/pretrained.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -44,14 +44,14 @@ pub static PREFAB_RESNET_MAP: StaticPreFabMap<ResNetContractConfig> = StaticPreF
4444
items: &[
4545
&StaticPretrainedWeightsDescriptor {
4646
name: "tv_in1k",
47-
description: "ResNet-18 pretrained on ImageNet",
47+
description: "TorchVision ResNet-18",
4848
license: Some("bsd-3-clause"),
4949
origin: Some("https://github.com/pytorch/vision"),
5050
urls: &["https://download.pytorch.org/models/resnet18-f37072fd.pth"],
5151
},
5252
&StaticPretrainedWeightsDescriptor {
5353
name: "a1_in1k",
54-
description: "ResNet-18 pretrained on ImageNet",
54+
description: "RSB Paper ResNet-18 a1",
5555
license: None,
5656
origin: Some("https://github.com/huggingface/pytorch-image-models"),
5757
urls: &[
@@ -60,7 +60,7 @@ pub static PREFAB_RESNET_MAP: StaticPreFabMap<ResNetContractConfig> = StaticPreF
6060
},
6161
&StaticPretrainedWeightsDescriptor {
6262
name: "a2_in1k",
63-
description: "ResNet-18 pretrained on ImageNet",
63+
description: "RSB Paper ResNet-18 a2",
6464
license: None,
6565
origin: Some("https://github.com/huggingface/pytorch-image-models"),
6666
urls: &[
@@ -69,7 +69,7 @@ pub static PREFAB_RESNET_MAP: StaticPreFabMap<ResNetContractConfig> = StaticPreF
6969
},
7070
&StaticPretrainedWeightsDescriptor {
7171
name: "a3_in1k",
72-
description: "ResNet-18 pretrained on ImageNet",
72+
description: "RSB Paper ResNet-18 a3",
7373
license: None,
7474
origin: Some("https://github.com/huggingface/pytorch-image-models"),
7575
urls: &[
@@ -88,14 +88,14 @@ pub static PREFAB_RESNET_MAP: StaticPreFabMap<ResNetContractConfig> = StaticPreF
8888
items: &[
8989
&StaticPretrainedWeightsDescriptor {
9090
name: "tv_in1k",
91-
description: "ResNet-34 pretrained on ImageNet",
91+
description: "TorchVision ResNet-34",
9292
license: Some("bsd-3-clause"),
9393
origin: Some("https://github.com/pytorch/vision"),
9494
urls: &["https://download.pytorch.org/models/resnet34-b627a593.pth"],
9595
},
9696
&StaticPretrainedWeightsDescriptor {
9797
name: "a1_in1k",
98-
description: "ResNet-34 pretrained on ImageNet",
98+
description: "RSB Paper ResNet-32 a1",
9999
license: None,
100100
origin: Some(
101101
"https://github.com/huggingface/pytorch-image-models/releases",
@@ -106,7 +106,7 @@ pub static PREFAB_RESNET_MAP: StaticPreFabMap<ResNetContractConfig> = StaticPreF
106106
},
107107
&StaticPretrainedWeightsDescriptor {
108108
name: "a2_in1k",
109-
description: "ResNet-34 pretrained on ImageNet",
109+
description: "RSB Paper ResNet-32 a2",
110110
license: None,
111111
origin: Some(
112112
"https://github.com/huggingface/pytorch-image-models/releases",
@@ -117,7 +117,7 @@ pub static PREFAB_RESNET_MAP: StaticPreFabMap<ResNetContractConfig> = StaticPreF
117117
},
118118
&StaticPretrainedWeightsDescriptor {
119119
name: "a3_in1k",
120-
description: "ResNet-34 pretrained on ImageNet",
120+
description: "RSB Paper ResNet-32 a3",
121121
license: None,
122122
origin: Some(
123123
"https://github.com/huggingface/pytorch-image-models/releases",
@@ -151,7 +151,7 @@ pub static PREFAB_RESNET_MAP: StaticPreFabMap<ResNetContractConfig> = StaticPreF
151151
// ERROR: Some<Downsample> stub cannot be applied to None
152152
&StaticPretrainedWeightsDescriptor {
153153
name: "tv_in1k",
154-
description: "ResNet-50 pretrained on ImageNet",
154+
description: "TorchVision ResNet-50",
155155
license: Some("bsd-3-clause"),
156156
origin: Some("https://github.com/pytorch/vision"),
157157
urls: &["https://download.pytorch.org/models/resnet50-0676ba61.pth"],

0 commit comments

Comments
 (0)