Assertion `input ＞= 0. && input ＜= 1.` failed._assertion input的值

作者：Cpp五条 | 2024-03-18 14:21:47

踩

assertion input的值

报错提示：
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
在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Cpp五条/article/detail/263671