runtime error

Downloading: "https://download.pytorch.org/models/vit_b_16-c867db91.pth" to /home/user/.cache/torch/hub/checkpoints/vit_b_16-c867db91.pth 0%| | 0.00/330M [00:00<?, ?B/s] 76%|███████▌ | 250M/330M [00:01<00:00, 262MB/s] 100%|██████████| 330M/330M [00:01<00:00, 267MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 13, in <module> vit.load_state_dict(torch.load(f="09_pretrained_vit_feature_extractor_pizza_steak_sushi_20_percent.pth", File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1604, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for VisionTransformer: Missing key(s) in state_dict: "heads.head.weight", "heads.head.bias", "classifier.1.weight", "classifier.1.bias". Unexpected key(s) in state_dict: "heads.0.weight", "heads.0.bias".

Container logs: