Assertion `input >= 0. && input <= 1.` failed.

报错提示:
Assertioninput >= 0. && input <= 1.failed.
可能原因是nn.BCELoss()的输入要是0到1的小数。而不是一个负数或者一个大于1的数字。有可能是因为你的激活函数的输出值不在0到1之间了。比如tanh,改成sigmoid可以解决。
即将return out 更改为 return F.sigmoid(out),或者改为out= nn.Sigmoid()(out), return out
Assertion `input >= 0. && input <= 1.` failed._第1张图片

你可能感兴趣的:(8种liver分割)