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)
...
原创
2021-04-28 21:03:36 ·
1124 阅读 ·
1 评论