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

fail on val #48

Open Qu1n-22 opened 2 months ago

Qu1n-22 commented 2 months ago

Did it fail because it was too inaccurate?

''' 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 60/60 [00:19<00:00, 3.13it/s] Traceback (most recent call last): File "/public2/home/wangchen/1U/VM-UNet-main/train.py", line 223, in main(config) File "/public2/home/wangchen/1U/VM-UNet-main/train.py", line 178, in main loss = val_one_epoch( File "/public2/home/wangchen/1U/VM-UNet-main/engine.py", line 91, in val_one_epoch TN, FP, FN, TP = confusion[0,0], confusion[0,1], confusion[1,0], confusion[1,1] IndexError: index 1 is out of bounds for axis 1 with size 1 '''