liwenyun
/
advclip
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 svhn_at.py

This commit is contained in:
liwenyun 2024-10-28 19:01:32 +08:00
parent e67581ad23
commit 888abd05b1
1 changed files with 0 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
svhn_at.py
View File

@ -29,16 +29,6 @@ IMAGENET_DEFAULT_STD = processor.image_std
size = processor.size["height"]
model = timm.create_model("timm/vit_base_patch16_224.orig_in21k_ft_in1k",
pretrained=False)
model.head = nn.Linear(model.head.in_features, 10)
model.load_state_dict(
torch.hub.load_state_dict_from_url(
"https://huggingface.co/edadaltocg/vit_base_patch16_224_in21k_ft_svhn/resolve/main/pytorch_model.bin",
map_location="cpu",
file_name="vit_base_patch16_224_in21k_ft_svhn.pth",
)
)
model = timm.create_model("timm/vit_base_patch16_224.orig_in21k_ft_in1k",
pretrained=False)