Skip to content
Snippets Groups Projects
Commit 578d4d0c authored by wangbin's avatar wangbin Committed by GitHub
Browse files

[Fix] Fix the bug in binary_cross_entropy (#1527)


* [Fix] Fix the bug in binary_cross_entropy

 Fix the bug in binary_cross_entropy
'label.max() <= 1' should mask out ignore_index, since the ignore_index often set as 255.

* [Fix] Fix the bug in binary_cross_entropy, add comments

As the ignore_index often set as 255, so the binary class label check should mask out ignore_index.

Co-authored-by: default avatarMiao Zheng <76149310+MeowZheng@users.noreply.github.com>

* [Fix] Fix the bug in binary_cross_entropy

As the ignore_index often set as 255, so the binary class label check should mask out ignore_index.

Co-authored-by: default avatarMiao Zheng <76149310+MeowZheng@users.noreply.github.com>
Co-authored-by: default avatarMeowZheng <meowzheng@outlook.com>
parent d0955901
No related branches found
No related tags found
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment