Skip to content

Direct Preference Optimization (DPO) style rewards#99

Merged
opentaco merged 12 commits intostagingfrom
feature/dpo-rewards
Aug 24, 2023
Merged

Direct Preference Optimization (DPO) style rewards#99
opentaco merged 12 commits intostagingfrom
feature/dpo-rewards

Conversation

@opentaco
Copy link
Contributor

Direct Preference Optimization (DPO) style rewards

Calculates a direct preference optimization (DPO) style reward for a completion, which is a reference model's average log-probability for completion tokens given a prompt.

Uses guidance from https://github.com/eric-mitchell/direct-preference-optimization/blob/main/trainers.py.

Copy link
Contributor

@steffencruz steffencruz left a comment

Choose a reason for hiding this comment

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

Interesting changes. Looking forward to seeing the experiments

@opentaco opentaco marked this pull request as ready for review August 24, 2023 08:36
@opentaco opentaco requested a review from Eugene-hu August 24, 2023 08:37
@opentaco opentaco merged commit 6286e9c into staging Aug 24, 2023
@opentaco opentaco deleted the feature/dpo-rewards branch August 24, 2023 15:32
@p-ferreira p-ferreira mentioned this pull request Aug 28, 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.

3 participants