A issues about ckpt

[18222why]

when I using --model vit_base_patch16 to train and mage-vitb-ft.pth as ckpt,A issues coming:

Traceback (most recent call last):
File "main_finetune.py", line 355, in
main(args)
File "main_finetune.py", line 250, in main
assert set(msg.missing_keys) == {'head.weight', 'head.bias', 'fc_norm.weight', 'fc_norm.bias'}
AssertionErro

what should i do?

[LTH14]

mage-vitb-ft.pth is an already fine-tuned checkpoint. The current fine-tuning code only supports fine-tuning a pre-trained checkpoint.

[18222why]

Thanks,I finshed it.

[18222why]

mage-vitb-ft.pth is an already fine-tuned checkpoint. The current fine-tuning code only supports fine-tuning a pre-trained checkpoint.

Dear professor, I now have a very strange problem. When I was running the fine-tuning, an unexpected problem occurred. The accuracy rate remained at 78.6 and would not be changed until 100epoch.But it has never happened before, and it will train 10 epochs before it will reach 78.6. Is my model overfitting？