Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Disable dropout in DPO Training #639

Merged
merged 7 commits into from
Aug 14, 2023

Conversation

NouamaneTazi
Copy link
Member

@NouamaneTazi NouamaneTazi commented Aug 11, 2023

Helps stabilising training.
For now, I only added disable_dropout to DPO Training, but let me know if you think it should go to DPOTrainer with a config to disable

cc @kashif

@NouamaneTazi NouamaneTazi changed the title disable dropout in dpo Disable dropout in DPO Training Aug 11, 2023
@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented Aug 11, 2023

The documentation is not available anymore as the PR was closed or merged.

@kashif
Copy link
Collaborator

kashif commented Aug 13, 2023

LGTM! lets add it to the trainer I would say...

@NouamaneTazi
Copy link
Member Author

@kashif Done!

@kashif
Copy link
Collaborator

kashif commented Aug 13, 2023

thank you so much!

@lvwerra lvwerra merged commit 98120d6 into huggingface:main Aug 14, 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