runtime error

Traceback (most recent call last): File "/home/user/app/app.py", line 19, in <module> v.load_state_dict(torch.load(ckpt_name, map_location=torch.device('cpu')).state_dict()) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2152, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for SimpleViT: Missing key(s) in state_dict: "transformer.norm.weight", "transformer.norm.bias", "linear_head.weight", "linear_head.bias". Unexpected key(s) in state_dict: "linear_head.0.weight", "linear_head.0.bias", "linear_head.1.weight", "linear_head.1.bias".

Container logs: