RWXMU 这个人很懒，什么都没有留下～ 1 文章 0 评论 0 粉丝 关注私信 文章 评论 问答 关注 收藏 Python pytorch交叉熵损失函数的weight参数的使用 首先 必须将权重也转为Tensor的cuda格式； 然后 将该class_weight作为交叉熵函数对应参数的输入值。 class_weight = torch.FloatTens… RWXMU 2025年1月4日 点击查看更多
