Error

[JackieWuuuuuu]

Traceback (most recent call last):
File "train.py", line 189, in
main(config)
File "train.py", line 132, in main
train_one_epoch(
File "/root/UltraLight-VM-UNet-main/engine.py", line 40, in train_one_epoch
loss.backward()
File "/root/miniconda3/envs/vmunet/lib/python3.8/site-packages/torch/_tensor.py", line 487, in backward
torch.autograd.backward(
File "/root/miniconda3/envs/vmunet/lib/python3.8/site-packages/torch/autograd/init.py", line 197, in backward
Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
用服务器跑，到Training的步骤就弹出这个报错，解决很久都无法解决，请问应该如何解决？

[wurenkai]

Hi, you may be able to get an answer to this question from issue 6, or from issue 4 of our other project.

[JackieWuuuuuu]

请问数据集这样放置是否正确？分别只有images和masks的。
我的Prepare_ISIC2017.py是这样

config_setting.py是这样

运行Prepare_ISIC2017.py会得到这样的结果

我认为是路径问题？您能帮我看看是什么问题吗？如果可以，十分感激！！

[wurenkai]

You are right, running Prepare_ISIC2017.py shows “[]” indicating that the image was not read correctly. You should make sure that the code Print(Tr_list) reads out the image naming. I see no problem with your current code, note that Prepare_ISIC2017.py is placed at the “/data” file directory level. Also, the image suffix inside images is jpg and inside masks the image is png, which is the suffix specified by the code so far.

In addition, we have provided code to prepare your own dataset, so you can take your own dataset (or substitute it for the ISIC2017 dataset), place it in the corresponding directory, and modify the number of training, validation, and test sets to run it. This will not have path problems.

[JackieWuuuuuu]

十分感谢指导！已成功跑通，感谢！

[JackieWuuuuuu]

作者您好，我想问问应该如何对自己特定的数据集进行测试呢

是只需要修改一下loader的中的路径就可以了吗（不用.npy文件测试，用自己的数据集测试）

在这里写入.pth的路径？如果您能帮助到我，十分感谢！