DRIVE

[zxt-triumph]

Traceback (most recent call last):
File "/home/dell/ZXT/Code/Segmenation/VAFF-Net/train-DRIVE.py", line 362, in
train_(myModel,dataloader_Train, dataloader_Test)
File "/home/dell/ZXT/Code/Segmenation/VAFF-Net/train-DRIVE.py", line 283, in train_
loss_det = criterion_dectect(outputs_det, labels_det)
File "/home/dell/anaconda3/envs/pytorch/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "/home/dell/ZXT/Code/Segmenation/VAFF-Net/utils/detectLoss.py", line 47, in forward
coo_pred = pred_tensor[coo_mask].view(-1,6) #提取所有有目标的区域
IndexError: The shape of the mask [1, 64, 64, 6] at index 3 does not match the shape of the indexed tensor [1, 64, 64, 4] at index 3

[figcommon]

How did you solve the problem？I have the same bug.