关于BCEWithLogitsLoss的bug?
收藏
神经网络的输出不带Sigmoid激活函数,使用BCEWithLogitsLoss损失函数会报错。但是再加上Sigmoid激活函数就能正常运行了。报错如下:
0
收藏
请登录后评论
提issue问吧