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

Fix saving tokenizers in DPR training + unify save and load dirs #682

Merged
merged 2 commits into from
Dec 16, 2020

Conversation

bogdankostic
Copy link
Contributor

Fixes #677 and #678.

This PR unifies the default directory names for saving and loading DPR models. Furthermore, query and passage tokenizer are now saved in the correct directory.

@bogdankostic
Copy link
Contributor Author

The tests fail with exit code 137, which seems to be an out-of-memory issue.

Copy link
Member

@tholor tholor left a comment

Choose a reason for hiding this comment

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

LGTM

@bogdankostic bogdankostic merged commit a9bcabc into master Dec 16, 2020
@bogdankostic bogdankostic deleted the save_DPR_models branch December 16, 2020 16:09
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.

Loading trained DPR retriever model gives error
2 participants