Closed mohitzsh closed 7 years ago
Hi, this repo works with python 2.x, I have not tested it with python 3.x.
Thanks for the reply! I’ll try with python 2.x.
Also, would it be okay if I ask you a few things about the implementation. I am trying to use your repo for adversarial training of semantic segmentation task and I am confused about a few things.
It's working with python 2.7.
But I get a runtime error
THCudaCheckFAIL file=/opt/conda/conda-bld/pytorch_1501972792122/work/pytorch-0.1.12/torch/lib/THC/generic/THCTensorCopy.c line=18 error=59 : device-side assert triggered Traceback (most recent call last): File "train.py", line 227, in
loss = loss + loss_calc(out[i+1],label[i+1],gpu0) File "train.py", line 128, in loss_calc label = Variable(label).cuda(gpu0) File "/home/sharm267/envs/pytorch-2.7/lib/python2.7/site-packages/torch/autograd/variable.py", line 243, in cuda return CudaTransfer(device_id, async)(self) File "/home/sharm267/envs/pytorch-2.7/lib/python2.7/site-packages/torch/autograd/_functions/tensor.py", line 160, in forward return i.cuda(async=self.async) File "/home/sharm267/envs/pytorch-2.7/lib/python2.7/site-packages/torch/_utils.py", line 65, in _cuda return newtype(self.size()).copy(self, async) RuntimeError: cuda runtime error (59) : device-side assert triggered at /opt/conda/conda-bld/pytorch_1501972792122/work/pytorch-0.1.12/torch/lib/THC/generic/THCTensorCopy.c:18
Any idea how I can handle this?
this error occurs when this instruction(given in readme) is not followed - how many labels are there in your data?
I am using the PascalVOC dataset. That's why I assumed I can work with the default value of noLabels.
to debug you could do this: print the max value of all labels for every batch. It should not be greater than 20, if you are using PASCAL VOC. In PASCAL VOC images, they have a background label, which is 255. I have merged it with background here.
Let me know if you issue is solved, I will close this issue.
I'll try to debug it tonight. I'll let you know how it goes.
You were spot on. One of the images had a label value 192.
Thanks again for the work. :D
hi, I am facing a similar issue, how do you check the number of labels of a ground truth image?
Thanks for the amazing work. I am trying to use your model description for another segmentation problem.
But when I run
python train.py
this is the error log that I get.
Additional Info: I am running python 3.5