FredZhang7

/

google-safesearch-mini-v2

like 3

model = timm.create_model("hf_hub:FredZhang7/google-safesearch-mini-v2", pretrained=True)

.....
RuntimeError: Error(s) in loading state_dict for InceptionResnetV2:
size mismatch for classif.weight: copying a param with shape torch.Size([2, 1536]) from checkpoint, the shape in current model is torch.Size([3, 1536]).
size mismatch for classif.bias: copying a param with shape torch.Size([2]) from checkpoint, the shape in current model is torch.Size([3]).

BUT

model = timm.create_model('inception_resnet_v2', pretrained=False, num_classes=3)
weights = torch.load('pytorch_model.bin')
model.load_state_dict(weights)

SUCCEEDED