Skip to content

Commit 211c0f5

Browse files
authored
Update convert_pytorch_to_paddle.rst (PaddlePaddle#1009)
1 parent 28a61b2 commit 211c0f5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/community/contribute_models/convert_pytorch_to_paddle.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,8 +164,8 @@ PaddleNLP(Paddle框架)和HuggingFace的transformers(PyTorch框架)里
164164
>>> import paddle
165165
>>> model_name = "bert-base-uncased"
166166
>>> # 模型下载地址: https://paddlenlp.bj.bcebos.com/models/transformers/bert-base-uncased.pdparams
167-
>>> model_file = "model_state.pdparams"
168-
>>> paddle_state_dict = torch.load(model_file)
167+
>>> model_file = "bert-base-uncased.pdparams"
168+
>>> paddle_state_dict = paddle.load(model_file)
169169
>>> paddle_state_dict.keys()
170170
dict_keys(['bert.embeddings.word_embeddings.weight', 'bert.embeddings.position_embeddings.weight', 'bert.embeddings.token_type_embeddings.weight',
171171
'bert.embeddings.layer_norm.weight', 'bert.embeddings.layer_norm.bias',

0 commit comments

Comments
 (0)