UltraLight-VM-UNet icon indicating copy to clipboard operation
UltraLight-VM-UNet copied to clipboard

How do I fix this error

Open TPF0628 opened this issue 10 months ago • 5 comments

ModuleNotFoundError: No module named 'selective_scan_cuda'

TPF0628 avatar Apr 22 '24 09:04 TPF0628

Hi, based on your reported error message, we believe there is an issue with the environment installation. We recommend that you follow our 'Main Environments' section to install the correct environment.

wurenkai avatar Apr 22 '24 12:04 wurenkai

When I train and test, I always get an error during the test step: "IndexError: index 1 is out of bounds for axis 1 with size 1." How can I fix this? Thank you.

TPF0628 avatar Apr 26 '24 07:04 TPF0628

Hi, I hope the issue will help you.

wurenkai avatar Apr 26 '24 10:04 wurenkai

Yeah thanks for your reply, I'm also guessing that the problem is in the data preparation stage. I am using my own prepared data set, images are 8-bit gray images, and masks are 8-bit binary images. 0001 0001

TPF0628 avatar Apr 27 '24 05:04 TPF0628

and num_classes still = 1

TPF0628 avatar Apr 27 '24 05:04 TPF0628