Skip to content

Commit d87de4d

Browse files
committed
feat: add new ResNet-50 pretrained weight descriptor (a1_in1k)
1 parent 88aa132 commit d87de4d

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

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

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,17 @@ pub static PREFAB_RESNET_MAP: StaticPreFabMap<ResNetContractConfig> = StaticPreF
162162
license: Some("bsd-3-clause"),
163163
origin: Some("https://github.com/pytorch/vision"),
164164
urls: &["https://download.pytorch.org/models/resnet50-11ad3fa6.pth"],
165+
},
166+
&StaticPretrainedWeightsDescriptor {
167+
name: "a1_in1k",
168+
description: "ResNet-50 pretrained on ImageNet",
169+
license: Some("bsd-3-clause"),
170+
origin: Some(
171+
"https://github.com/huggingface/pytorch-image-models/releases",
172+
),
173+
urls: &[
174+
"https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet50_a1_0-14fe96d1.pth",
175+
],
165176
},
166177
*/
167178
],

0 commit comments

Comments
 (0)