JCruan519/VM-UNet

Pre-training on the decoder part of VMUnet

Opened this issue · 2 comments

CYYJL commented

Hi, Thank you very much for your work , Is there any pre-training weight in the decoder part of VMUnet

JCruan519 commented

@CYYJL
Hello, for the decoder part, we also initialize with pre-trained weights from VMamba, a method also used in Swin-UNet.

CYYJL commented

@CYYJL Hello, for the decoder part, we also initialize with pre-trained weights from VMamba, a method also used in Swin-UNet.

HI，Thank you very much for your advice，I konw how to do it.
Thank you very much!
Best!
CYYJL