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

为什么保存的模型权重不能再次用到训练中呢 #28

Open xiachashuanghua opened 6 months ago

xiachashuanghua commented 6 months ago

我使用保存的权重和预训练权重作比对，内容不一致，该怎么解决呢

JCruan519 commented 6 months ago

@xiachashuanghua Hello! Which two sets of weights are you referring to? The pretrained weights are those obtained from training VMamba on ImageNet. The weights provided in the results are the ones that have been fine-tuned on downstream tasks, so they are definitely different.

xiachashuanghua commented 6 months ago

训练完成后保存的best.pth无法在模型中当作预训练模型使用，这个问题怎么解决呢

@xiachashuanghua Hello! Which two sets of weights are you referring to? The pretrained weights are those obtained from training VMamba on ImageNet. The weights provided in the results are the ones that have been fine-tuned on downstream tasks, so they are definitely different.

sky1talent commented 5 months ago

确实遇到了这种问题，就是自己保存训练后的权重，报错显示没有模型，不能用 @JCruan519

sakurarma commented 5 months ago

@sky1talent @sky1talent 直接load你保存的权重，注释掉加载预训练模型的代码不行吗？我自己的直接加载没问题的

heguibing0413 commented 4 months ago

请问问题解决了吗，我用保存的权重训练TPFP一直是0