Skip to content

[TF] from_pt should respect authorized_unexpected_keys#8056

Merged
sshleifer merged 1 commit intohuggingface:masterfrom
sshleifer:respect-tf-authorized_unexpected_keys
Oct 26, 2020
Merged

[TF] from_pt should respect authorized_unexpected_keys#8056
sshleifer merged 1 commit intohuggingface:masterfrom
sshleifer:respect-tf-authorized_unexpected_keys

Conversation

@sshleifer
Copy link
Copy Markdown
Contributor

This comes in handy when

        "model.encoder.embed_tokens.weight",
        "model.decoder.embed_tokens.weight",

are in the PT state dict but not the TF symbolic weights.

Copy link
Copy Markdown
Collaborator

@sgugger sgugger 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 adding!

@sshleifer sshleifer merged commit bc9332b into huggingface:master Oct 26, 2020
@sshleifer sshleifer deleted the respect-tf-authorized_unexpected_keys branch October 26, 2020 17:53
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.

3 participants