mage icon indicating copy to clipboard operation
mage copied to clipboard

A issues about ckpt

Open 18222why opened this issue 5 months ago • 2 comments

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?

18222why avatar Sep 22 '24 02:09 18222why