"...resnet50_tensorflow.git" did not exist on "72dc30ea04accb87121e18c45410e6d2d2e040be"
Unverified Commit 92cfe8b0 authored by Arthur's avatar Arthur Committed by GitHub
Browse files

Remove `authorized_missing_keys`in favor of _keys_to_ignore_on_load_missing (#20228)

parent 2d920010
......@@ -246,8 +246,8 @@ class ConvBertPreTrainedModel(PreTrainedModel):
load_tf_weights = load_tf_weights_in_convbert
base_model_prefix = "convbert"
supports_gradient_checkpointing = True
authorized_missing_keys = [r"position_ids"]
authorized_unexpected_keys = [r"convbert.embeddings_project.weight", r"convbert.embeddings_project.bias"]
_keys_to_ignore_on_load_missing = [r"position_ids"]
_keys_to_ignore_on_load_unexpected = [r"convbert.embeddings_project.weight", r"convbert.embeddings_project.bias"]
def _init_weights(self, module):
"""Initialize the weights"""
......
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