热门标签
当前位置: article > 正文
nn.CrossEntropyLoss_bceloss typeerror: forward() got an unexpected key
作者：AllinToyou | 2024-03-22 08:06:55
赞
踩
bceloss typeerror: forward() got an unexpected keyword argument 'reduction
刚开始报错
TypeError: forward() got an unexpected keyword argument 'reduction'
- 1
查了半天后发现是写错地方了，原来是这么写的
criterion = nn.CrossEntropyLoss(weight=weights)
loss = criterion(x, y, reduction='sum'- 1
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/AllinToyou/article/detail/286470
推荐阅读
相关标签
