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
393 stars 12 forks source link

training results #45

Open JAYCHOU2020 opened 2 months ago

JAYCHOU2020 commented 2 months ago

Hi, thank you for your work while I was training at 4090 Single card, raining isic18 with the pretraining model pre_trained_weights/vssmsmall_dp03_ckpt_epoch_238.pth could not get the miou result of 81.35 in the paper.Only 0.793 miou

fangxiao-yu commented 2 months ago

Hello, can your loss go down? My loss can't go down. I don't know why

Aaronwangz commented 2 months ago

/vssmsmall_dp03_ckpt_epoch_238.pth，Was this weight found in vmanba? At that specific location?