Dear friends,
I’m trying to apply the segmentation notebook to the skin lesion segmentation problem (Skin Lesion Analysis Towards Melanoma Detection - ISIC 2016 challenge - https://challenge.kitware.com/#challenge/560d7856cad3a57cfde481ba), but I’m stuck on two problems:
- RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension […]
- RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/THCCachingHostAllocator.cpp:265
I’ve been trying to find out a solution but without any progress. When error 1) happens, I change the size of the images and get error 2 (someone said that the problem can be the size - it must be even)
My notebook:
https://colab.research.google.com/drive/1s2XgYDT5XyAGoPwDardlELQ1aYvZns5X
I changed the accuracy to this:
def seg_accuracy(input, target):
target = target.squeeze(1)
return (input.argmax(dim=1)==target).float().mean()
and also tried to use only one transform for data augmentation, but the errors are still with me
Anyone knows what is going on?
Thanks in advance!
Leo