inceptionnext icon indicating copy to clipboard operation
inceptionnext copied to clipboard

A error occurred when I load the pretrained weights

Open leij0318 opened this issue 1 year ago • 2 comments

When I run thre code below:

model = inceptionnext_base(pretrained=True) inputs = torch.randn((1, 3, 640, 640)) for i in model(inputs): print(i.size())

RuntimeError: Error(s) in loading state_dict for MetaNeXt: Unexpected key(s) in state_dict: "head.fc1.weight", "head.fc1.bias", "head.norm.weight", "head.norm.bias", "head.fc2.weight", "head.fc2.bias".

leij0318 avatar Oct 19 '23 02:10 leij0318