Paddle the gpu kernel of cross entropy with soft label can be optimized by using reduce.

Paddle PaddlePaddle-Gardener 2022年5月19日 05:20 12 查看原文

源自github用户lcy-seso: The current implementation of the GPU kernel of the cross_entropy op with the soft label can be further optimized by using reduce: https://github.com/PaddlePaddle/Paddle/blob/develop/paddle/operators/cross_entropy_op.cu#L38

This problem is also known by @qingqing01

回答
2 条回答

源自github用户typhoonzero: I used thrust::reduce for convenience, it's very simple to use and efficient: https://github.com/PaddlePaddle/Paddle/blob/develop/paddle/operators/accuracy_op.cu#L43

热门问题
相关推荐
推荐分类