首页 Paddle框架 帖子详情
paddlepaddle 1.5.0 cross_entropy对不齐 其他框架
收藏
快速回复
Paddle框架 问答深度学习 1034 1
paddlepaddle 1.5.0 cross_entropy对不齐 其他框架
收藏
快速回复
Paddle框架 问答深度学习 1034 1

这是结论图:softmax+cross_entropy 对不上pytorch的nn.CrossEntropyLoss,而且有明显的特点就是大于64的loss都变为64;图中分别是cross_entropy 和 nn.CrossEntropyLoss的输出
8741d8d974eb9eb0f28e64fc9

这是特征的输入100x2:
score_out_tr_np.npy.zip

这是上面数据通过paddle1.5.0的softmax层之后的结果:
softmax_out_tr_np.npy.zip

这是分类的真实标签100x1:
label_tr_np.npy.zip

这是paddle1.5.0的cross_entropy结果:
ce_tr_np.npy.zip

这是pytorch 0.4.1的nn.CrossEntropyLoss(reduce=False)的结果

loss_fn = nn.CrossEntropyLoss(reduce=False)
tmp_np = loss_fn(torch.from_numpy(score_out_tr_np), torch.from_numpy(label_tr_np[:, 0]))

torch_res.npy.zip

0
收藏
回复
全部评论(1)
时间顺序
aistudio_2
#3 回复于2021-05

mark

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户