Pretrained model not compatible with downstream training

I am having some issues using a private pretrained encoder weights. When following the swin-unetr pretrain repo to train a pretrained encoder weights, the model can't be loaded for downstream segmentation training. Here is the error:

KeyError: 'module.patch_embed.proj.weight'

Is there anything that might be missing from the pretrain repo? Would it be possible to fix this issue without retraining? Thank you!!

Inizia con il swin-unetr pretrain repo e il percorso di training della segmentazione downstream, quindi traccia come vengono caricati i pesi privati preaddestrati dell’encoder quando si verifica il KeyError per module.patch_embed.proj.weight. Il lavoro è completo quando si determina se i pesi preaddestrati possono essere caricati per la segmentazione downstream senza ripetere il training, oppure si documenta il requisito di compatibilità mancante.