lab-smile / GRACE

GNU General Public License v3.0
3 stars 0 forks source link

Pretrained Model can't load? #9

Open HeConnor opened 2 months ago

HeConnor commented 2 months ago

Hello,

I got error when I load pretrained model:

monai: v0.8.1

Error(s) in loading state_dict for UNETR:
    Missing key(s) in state_dict: "vit.patch_embedding.cls_token", "encoder2.blocks.0.1.conv3.conv.weight", "encoder2.blocks.1.1.conv3.conv.weight", "encoder3.blocks.0.1.conv3.conv.weight". 

Any help is welcome.

Thanks !