search

LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.
2.68k stars 595 forks source link

how to solve ? #5

Closed YangBai1109 closed 5 years ago

YangBai1109 commented 5 years ago

my label is background to 0 and others to 1 .but i donot know how to make it work, the code may not have a test function?

LeeJunHyun commented 5 years ago

which test function do you mean? If you have some problem with your data, then I recommend you to test this code with ISIC data.

LeeJunHyun commented 5 years ago

And you can refer this issue

YangBai1109 commented 5 years ago

ok,i will see . now i have another question Traceback (most recent call last): File "/home/panpan/wenjian/master/main.py", line 101, in <module> main(config) File "/home/panpan/wenjian/master/main.py", line 56, in main solver = Solver(config, train_loader, valid_loader, test_loader) File "/home/panpan/wenjian/master/solver.py", line 50, in __init__ self.device = torch.device('cuda'if torch.cuda.is_available() else 'cpu') AttributeError: module 'torch' has no attribute 'device' i donot know how to solve it

LeeJunHyun commented 5 years ago

Check your pytorch version and refer here :)

LeeJunHyun commented 5 years ago

If you don't have further questions, I will close the issue. I hope my answer was helpful.

YangBai1109 commented 5 years ago

thank you