guli-7721

有哪位大佬知道这个问题怎么解决 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

I solved it, if you put your own dataset and you need to change the dataset name, remember that it's in load_dataset_transforms.py, flare change it to your own dataset name....