File tree Expand file tree Collapse file tree 1 file changed +11
-0
lines changed
crates/bimm/src/models/resnet Expand file tree Collapse file tree 1 file changed +11
-0
lines changed Original file line number Diff line number Diff 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 ] ,
You can’t perform that action at this time.
0 commit comments