训练出错RuntimeError: ACL stream synchronize failed.

[xiaofei624]

Traceback (most recent call last):
File "train.py", line 195, in
amp=args.amp)
File "train.py", line 98, in train_net
multiclass=True)
File "/home/ma-user/work/xff/output/Pytorch-UNet-master_msft/utils/dice_score.py", line 42, in dice_loss
return 1 - fn(input, target, reduce_batch_first=True)
File "/home/ma-user/work/xff/output/Pytorch-UNet-master_msft/utils/dice_score.py", line 33, in multiclass_dice_coeff
dice += dice_coeff(input[:, channel, ...], target[:, channel, ...], reduce_batch_first, epsilon)
File "/home/ma-user/work/xff/output/Pytorch-UNet-master_msft/utils/dice_score.py", line 16, in dice_coeff
if sets_sum.item() == 0:
RuntimeError: ACL stream synchronize failed.