mxnet & pytorch 交叉熵损失

1. 对比 pytorch: nn.CrossEntropyLoss() & mxnet: gluon.loss.SoftmaxCrossEntropyLoss() 一般而言默认用法即可：(inputs只是分类器的得分，softmax和log包括在loss里) pytorch： d=np.array
posted @ 2020-06-22 16:07  三年一梦  阅读(621)  评论(0)  编辑  收藏  举报