UltraLight-VM-UNet
UltraLight-VM-UNet copied to clipboard
How do I fix this error
ModuleNotFoundError: No module named 'selective_scan_cuda'
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.
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.
Hi, I hope the issue will help you.
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.
and num_classes still = 1