unable to get repr for ＜class ‘torch.Tensor‘＞错误可能原因

自己在做语义分割的时候，出现了这个问题：unable to get repr for ＜class ‘torch.Tensor‘＞

自己使用了pycharm进行了模型训练，错误提示为：RuntimeError: CUDA error: device-side assert triggered

根据错误提示信息找到了出错位置在损失函数使用的地方，在训练的损失函数位置加了断点，进行调试，发现了 unable to get repr for ＜class ‘torch.Tensor‘＞ 的错误信息。

我出错的原因是由于，放进BCE损失函数的tensor没有进行（0-1）之间的映射。由于做的是二分类，所以将input进行sigmoid处理即可。

如图所示，一开始我错误语句为:

loss = self.criterion(SR,GT)

正确的语句为：

loss = self.criterion(SR_probs,GT)