Skip to content

Commit d87d119

Browse files
committed
Enable "resnet18:a1_in1k" on new 0.19.0 pytorch loader.
1 parent 07c85d1 commit d87d119

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

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

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,6 @@ pub static PREFAB_RESNET_MAP: StaticPreFabMap<ResNetContractConfig> = StaticPreF
4949
origin: Some("https://github.com/pytorch/vision"),
5050
urls: &["https://download.pytorch.org/models/resnet18-f37072fd.pth"],
5151
},
52-
/*
53-
DeserializeError("Candle Tensor error: invalid Zip archive: Could not find central directory end")
5452
&StaticPretrainedWeightsDescriptor {
5553
name: "a1_in1k",
5654
description: "ResNet18 pretrained on ImageNet",
@@ -60,7 +58,6 @@ pub static PREFAB_RESNET_MAP: StaticPreFabMap<ResNetContractConfig> = StaticPreF
6058
"https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet18_a1_0-d63eafa0.pth",
6159
],
6260
},
63-
*/
6461
],
6562
}),
6663
},

0 commit comments

Comments
 (0)