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 ·
1013 阅读 ·
1 评论