my input dimension is (2,1,116,132,132) my target dimension is (2,1,28,44,44)
i seem to be getting this error!
loss Traceback (most recent call last):
File "train3DUNet.py", line 143, in <module>
gpu=options.gpu)
File "train3DUNet.py", line 94, in train_net
print ('loss' ,loss)
File "/usr/local/lib/python3.5/dist-packages/torch/autograd /variable.py", line 119, in __repr__
return 'Variable containing:' + self.data.__repr__()
File "/usr/local/lib/python3.5/dist-packages/torch/tensor.py", line 133, in __repr__
return str(self)
File "/usr/local/lib/python3.5/dist-packages/torch/tensor.py", line 140, in __str__
return _tensor_str._str(self)
File "/usr/local/lib/python3.5/dist-packages/torch/_tensor_str.py", line 295, in _str
strt = _vector_str(self)
File "/usr/local/lib/python3.5/dist-packages/torch/_tensor_str.py", line 271, in _vector_str
fmt, scale, sz = _number_format(self)
File "/usr/local/lib/python3.5/dist-packages/torch/_tensor_str.py", line 79, in _number_format
tensor = torch.DoubleTensor(tensor.size()).copy_(tensor).abs_().view(tensor.nelement())
RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/torch/lib/THC/generic/THCTensorCopy.c:70
Im not really sure what to make of this ,My Gpu has enough memory,so i dont think this is a memory issue,if any one has any suggestions it would be really helpful,Thanks in advance !