3DUNet-Pytorch

3DUNet-Pytorch copied to clipboard

维度不匹配问题

5

有哪位大佬知道这个问题怎么解决 dice += (pred[:,i] * target[:,il)sum(din=l)sum(din=l)sun(din=l) / ((pred[:,i] * target[:,i)sum(din=l)sum(din=l).sumn(din=l)+ RuntimeError: The size of tensor a (1024) must match the size of tensor b (512) at non-sinaleton dimension 3

guli-7721

我要训练我自己的数据，为MRI格式，需要大量修改数据预处理的代码和训练代码吗？

CLGRY

维度问题

5

File "/opt/data/private/3DUNet-Pytorch-master/dataset/transforms.py", line 153, in __call__ img, mask = t(img, mask) File "/opt/data/private/3DUNet-Pytorch-master/dataset/transforms.py", line 64, in __call__ tmp_img[:,:es-ss] = img[:,ss:es] RuntimeError: The expanded size of the tensor (48) must match...

Thehonst

RuntimeError: CUDA out of memory. Tried to allocate 38.81 GiB (GPU 0; 48.00 GiB total capacity; 21.21 GiB already allocated; 23.81 GiB free; 22.30 GiB reserved in total by PyTorch)...

Bihuaa