Skip to content

[Unity][Training] Categorical cross entropy loss#14757

Merged
echuraev merged 1 commit intoapache:unityfrom
ndemashov:mlc-dev/categorical_cross-entropy_loss
May 12, 2023
Merged

[Unity][Training] Categorical cross entropy loss#14757
echuraev merged 1 commit intoapache:unityfrom
ndemashov:mlc-dev/categorical_cross-entropy_loss

Conversation

@ndemashov
Copy link
Contributor

This PR adds support of categorical cross entropy loss

@tvm-bot
Copy link
Collaborator

tvm-bot commented May 3, 2023

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

Generated by tvm-bot

@ndemashov ndemashov changed the base branch from main to unity May 3, 2023 13:09
@echuraev echuraev changed the title Categorical cross entropy loss [Unity][Training] Categorical cross entropy loss May 4, 2023
Copy link
Contributor

@echuraev echuraev left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you for the PR. Please fix issues in CI. I added a minor comment. @Ubospica, @SiriusNEO could you please review this PR?

SiriusNEO
SiriusNEO approved these changes May 4, 2023
Copy link
Contributor

@SiriusNEO SiriusNEO left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Overall looks great, just a few comments.

@ndemashov ndemashov force-pushed the mlc-dev/categorical_cross-entropy_loss branch from c370d7f to 63d0cc7 Compare May 4, 2023 11:40
@ndemashov ndemashov force-pushed the mlc-dev/categorical_cross-entropy_loss branch from 63d0cc7 to 4edda8a Compare May 4, 2023 12:15
Copy link
Contributor

@echuraev echuraev left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@echuraev
Copy link
Contributor

echuraev commented May 4, 2023

@tvm-bot rerun

@ndemashov ndemashov force-pushed the mlc-dev/categorical_cross-entropy_loss branch 6 times, most recently from 0674bec to 03d625c Compare May 7, 2023 20:47
@echuraev
Copy link
Contributor

@tvm-bot rerun

1 similar comment
@echuraev
Copy link
Contributor

@tvm-bot rerun

fix lint

Add targets dtype checker and description of the reason for two implementations

Lift the common part

fix dtype checker

fix lint
@ndemashov ndemashov force-pushed the mlc-dev/categorical_cross-entropy_loss branch from 03d625c to 775edc9 Compare May 11, 2023 07:57
@echuraev echuraev merged commit 25a491d into apache:unity May 12, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants