A question about training: Not loaded keys

JCruan519 / VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".

Apache License 2.0

393 stars 12 forks source link

A question about training: Not loaded keys #18

Open tobedailao opened 3 months ago

tobedailao commented 3 months ago

When I'm training, I encounter some issues as shown in the next figure. It seems that certain parameters don't have corresponding weights, so they are going to be trained from scratch. This consumes a lot of time. How can I handle this question?

liubaoning111 commented 3 months ago

hello，have you solved the problem？

tobedailao commented 3 months ago

hello，have you solved the problem？

hello！no. this question may not be solved

JCruan519 commented 3 months ago

@liubaoning111 @tobedailao

Hello, this situation is normal because VM-UNet only utilizes partial pre-trained weights.