Open startwang23 opened 10 months ago
Traceback (most recent call last): File "D:\unet3\UNet3Plus-main\UNet3Plus-main\train.py", line 494, in flag=fit_one_epoch(model_train, model, loss_history, eval_callback, optimizer, epoch, File "D:\unet3\UNet3Plus-main\UNet3Plus-main\utils\utils_fit.py", line 89, in fit_one_epoch main_dice = Dice_loss(outputs, pngs) File "D:\unet3\UNet3Plus-main\UNet3Plus-main\nets\UNet3Plus_training.py", line 286, in Dice_loss nt, ht, wt, ct = target.size() ValueError: not enough values to unpack (expected 4, got 3) Epoch 1/120: 0%| | 0/15 [00:02<?, ?it/s<class 'dict'>]
我也有一样的问题
yddjh
commented
6 months ago yddjh
commented
6 months ago
Traceback (most recent call last): File "D:\unet3\UNet3Plus-main\UNet3Plus-main\train.py", line 494, in
flag=fit_one_epoch(model_train, model, loss_history, eval_callback, optimizer, epoch,
File "D:\unet3\UNet3Plus-main\UNet3Plus-main\utils\utils_fit.py", line 89, in fit_one_epoch
main_dice = Dice_loss(outputs, pngs)
File "D:\unet3\UNet3Plus-main\UNet3Plus-main\nets\UNet3Plus_training.py", line 286, in Dice_loss
nt, ht, wt, ct = target.size()
ValueError: not enough values to unpack (expected 4, got 3)
Epoch 1/120: 0%| | 0/15 [00:02<?, ?it/s<class 'dict'>]