p85971406
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!47692 BCEGrad bugfix for loss calculation when reduction is Mean 

Merge pull request !47692 from zhangyanhui/code_mas
This commit is contained in:
i-robot 2023-01-10 08:22:31 +00:00 committed by Gitee
parent d4c414dac0 f67b3f69fc
commit 560e83bad0
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/binary_cross_entropy_grad.c
View File

@ -37,7 +37,7 @@ int BinaryCrossEntropyGrad(const int input_size, const int reduction, const floa
} }
} else { } else {
float dloss1 = dloss[0]; float dloss1 = dloss[0];
if (reduction == Reduction_Sum) { if (reduction == Reduction_Mean) {
dloss1 = dloss[0] / input_size; dloss1 = dloss[0] / input_size;
} }
for (int i = 0; i < input_size; i++) { for (int i = 0; i < input_size; i++) {