*** RuntimeError: Unknown model (resnet18_cifar10)
My timm version 0.6.13
(Pdb) model = timm.create_model("resnet18_cifar10", pretrained=pretrained)
*** RuntimeError: Unknown model (resnet18_cifar10)
(Pdb) model = timm.create_model("resnet18_cifar10", pretrained=False)
*** RuntimeError: Unknown model (resnet18_cifar10)
What to do to find this model?
model = timm.create_model("hf_hub:edadaltocg/resnet18_cifar10", pretrained=True)
Error(s) in loading state_dict for ResNet:
size mismatch for conv1.weight: copying a param with shape torch.Size([64, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([64, 3, 7, 7]).
I am also getting the same error that lumalav is getting!
Error(s) in loading state_dict for ResNet:
size mismatch for conv1.weight: copying a param with shape torch.Size([64, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([64, 3, 7, 7]).
Hi @lumalav @ttw , maybe you could refer to another issues in the repo for help, I've successfully setup the model with the guidance in it. :) https://huggingface.co/edadaltocg/resnet18_cifar10/discussions/1#643d1f65455be6574c82b093