search

JCruan519 / VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".
Apache License 2.0
471 stars 22 forks source link

Total model_dict: 222, Total pretrained_dict: 446, update: 117 Not loaded keys: ['layers.2.blocks.2.ln_1.weight', #23

Open Xi-Mu-Yu opened 6 months ago

Xi-Mu-Yu commented 6 months ago

HX48~%T2}XQ{WT@_XS8GHD6 When I was reproducing the paper, this question appeared, I don't know if this is normal, I look forward to your answer!

JCruan519 commented 6 months ago

@BeiZhiRen Hello, this is normal, as VM-UNet only utilizes a portion of VMamba's weights for initialization.

Xi-Mu-Yu commented 6 months ago

@BeiZhiRen Hello, this is normal, as VM-UNet only utilizes a portion of VMamba's weights for initialization.

I think so too, thank you very much