最近,再修改交叉熵损失函数时,碰到这个错误:

这个问题的原因:传入的class_weight函数类型是list类型,导致报错
解决方法:将class_weight改为tensor类型。
本文探讨了在修改深度学习模型中遇到的关于class_weight参数为list类型引发的错误,并提供了将其转换为tensor类型的解决方案,确保模型训练顺利进行。
最近,再修改交叉熵损失函数时,碰到这个错误:

这个问题的原因:传入的class_weight函数类型是list类型,导致报错
解决方法:将class_weight改为tensor类型。

被折叠的 条评论
为什么被折叠?