.rename_state_dict_variables | R Documentation |
There are some hard-to-avoid differences between the variable names in BERT models constructed using this package and the standard variable names used in the Huggingface saved weights. This function changes the names from the Huggingface saves to match package usage.
.rename_state_dict_variables(state_dict)
state_dict |
A state_dict of pretrained weights, probably loaded from a file. |
The state_dict with the names normalized.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.