Skip to content

[ORPO] add nll_target for orpo nll loss#503

Merged
austin362667 merged 11 commits into
linkedin:mainfrom
kashif:orpo-nll-loss
Jan 9, 2025
Merged

[ORPO] add nll_target for orpo nll loss#503
austin362667 merged 11 commits into
linkedin:mainfrom
kashif:orpo-nll-loss

Conversation

@kashif

@kashif kashif commented Dec 29, 2024

Copy link
Copy Markdown
Contributor

Summary

add optional nll_target argument to calculate nll (needed for ORPO nll loss)

  • Hardware Type:
  • run make test to ensure correctness
  • run make checkstyle to ensure code style
  • run make test-convergence to ensure convergence

austin362667
austin362667 previously approved these changes Dec 31, 2024

@austin362667 austin362667 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

All tests passed, thank you! @kashif

@austin362667 austin362667 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM! Thank you @kashif

@austin362667 austin362667 merged commit 9586a87 into linkedin:main Jan 9, 2025
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.

2 participants