cannot load all 22k_models

[Nise-2-meet-U]

C:\ProgramData\Anaconda3\envs\yolo\python.exe G:/nft/code/inference.py
Traceback (most recent call last):
File "G:/nft/code/inference.py", line 37, in
model = con.convnext_xlarge(pretrained=True, in_22k=True)
File "G:\nft\code\convnext.py", line 211, in convnext_xlarge
model.load_state_dict(checkpoint["model"])
File "C:\ProgramData\Anaconda3\envs\yolo\lib\site-packages\torch\nn\modules\module.py", line 1482, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for ConvNeXt:
size mismatch for head.weight: copying a param with shape torch.Size([21841, 2048]) from checkpoint, the shape in current model is torch.Size([1000, 2048]).
size mismatch for head.bias: copying a param with shape torch.Size([21841]) from checkpoint, the shape in current model is torch.Size([1000]).

[theguardsgod]

u need to modify num_classes=21841

[kalelpark]

model = convnext_base(pretrained = True, in_22k = True, num_classes=21841)