Skip to content

GitLab

Unverified Commit c149d366 authored by Arthur's avatar Arthur Committed by GitHub
Browse files

add _keys_to_ignore_on_load_unexpected = [r"pooler"] (#20210)

parent 8dcf494e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/transformers/models/dpr/modeling_dpr.py
View file @ c149d366
......@@ -299,6 +299,7 @@ class DPRPretrainedContextEncoder(DPRPreTrainedModel):
load_tf_weights = None
base_model_prefix = "ctx_encoder"
_keys_to_ignore_on_load_missing = [r"position_ids"]
_keys_to_ignore_on_load_unexpected = [r"pooler"]
class DPRPretrainedQuestionEncoder(DPRPreTrainedModel):
......@@ -311,6 +312,7 @@ class DPRPretrainedQuestionEncoder(DPRPreTrainedModel):
load_tf_weights = None
base_model_prefix = "question_encoder"
_keys_to_ignore_on_load_missing = [r"position_ids"]
_keys_to_ignore_on_load_unexpected = [r"pooler"]
class DPRPretrainedReader(DPRPreTrainedModel):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment