首页 PaddleNLP 帖子详情
如何将本地数据传入paddle.nn.embedding的参数矩阵中? 已解决
收藏
快速回复
PaddleNLP 其他学习资料数据 654 1
如何将本地数据传入paddle.nn.embedding的参数矩阵中? 已解决
收藏
快速回复
PaddleNLP 其他学习资料数据 654 1

导入自己预训练的词向量,需要将本地数据导入到paddle.nn.embedding的参数矩阵中。可将本地词典向量读取为NumPy数据格式,然后使用paddle.nn.initializer.Assign这个API初始化paddle.nn.embedding里的param_attr参数,即可实现加载用户自定义(或预训练)的Embedding向量。

DeepGeGe
已解决
2# 回复于2021-12
paddle.nn.initializer.Assign不仅仅可以为embedding层参数赋值,其实可以给paddle搭建的网络中任意参数赋值。
0
收藏
回复
全部评论(1)
时间顺序
DeepGeGe
#2 回复于2021-12

paddle.nn.initializer.Assign不仅仅可以为embedding层参数赋值,其实可以给paddle搭建的网络中任意参数赋值。

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户