RuntimeError reduce failed to synchronize cudaErrorAssert device-side assert triggered

reduce failed to synchronize: cudaErrorAssert: device-side assert triggered

运行基于Unet的CrackForest代码时，出现该错误。

查阅https://blog.csdn.net/xiaoxifei/article/details/89328596，发现是因为BCELOSS损失函数输入tensor超1.0了，https://blog.csdn.net/weixin_43846347/article/details/106538862博客中的将BCELOSS更换为交叉熵LOSS，但是 我在我demo中更改为CrossEntropyLoss后报如下错

遂又将数据的label，改为：

还是报错：

label输入维度不匹配，为了确认是否是long（）修改了数据维度，打印出了inputs维度发现也是4D的，遂又回过头审视损失函