利用ｓｅｇｎｅｔ做图像分割～Ｂｕｇ

首先数据预处理确认是正确的，在ｕｎｅｔ模型上已经运行成功，当数据集在ｓｅｇｎｅｔ模型上时，会出现如下错误：

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

百度说是标签的ｉｎｄｅｘ出错，应该是从０开始的，但是我在读数据时，它就是自动从零开始，所以这个原因不可行。

后来检查代码发现模型的输出没有用激活函数

torch.nn.Sigmoid()(x11d)

在模型的最后添加这个就成功解决啦！