Skip to content

A issues about ckpt #56

@18222why

Description

@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?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions