Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

[BB2] Memory Only Fix #3950

Merged
merged 2 commits into from
Aug 18, 2021
Merged

[BB2] Memory Only Fix #3950

merged 2 commits into from
Aug 18, 2021

Conversation

klshuster
Copy link
Contributor

Patch description
We don't need to classify retrieval when we're dealing with --knowledge-access-method <memory_only>/<none>. We do not set the query_generator_vec unless we're in the other three regimes (search_only, classify, or all), and thus we should only generate the queries in these regimes as well.

Testing steps
Verified that parlai i -mf zoo:blenderbot2/blenderbot2_400M/model works with --knowledge-access-method memory_only (this would lead to a bug before)

Copy link
Contributor

@jxmsML jxmsML left a comment

Choose a reason for hiding this comment

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

LGTM! thanks for the fix

@klshuster klshuster merged commit cdbcd0e into master Aug 18, 2021
@klshuster klshuster deleted the bb2_memory_only branch August 18, 2021 19:14
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants