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

The loss became nan #21

Open qiuluyi opened 6 months ago

qiuluyi commented 6 months ago

4342a9cfd1305be0fefafb15ebbb9ce

JCruan519 commented 6 months ago

@qiuluyi Hello, could you provide more detailed information? For example, what configuration are you using, on which dataset are you training, and so on.

qiuluyi commented 6 months ago

I just use vmunet as the baseline and train it on the BraTS dataset, I don't change the configuration of vmunet