Skip to content

Direct Preference Optimization (DPO) style rewards - #99

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

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

Conversation

@opentaco

Copy link
Copy Markdown
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.

@steffencruz steffencruz 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.

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