Skip to content

X
X2Paddle

PaddlePaddle / X2Paddle
大约 1 年 前同步成功

通知 328
Star 698
Fork 167
X
X2Paddle
提交 ac9c0ead 编写于 作者: C Channingss
浏览文件
操作

modify embeding weight

上级 17cf4a57
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py
浏览文件 @ ac9c0ead
...@@ -721,7 +721,7 @@ class OpSet9(): ...@@ -721,7 +721,7 @@ class OpSet9():
op_name = name_generator("embedding", self.nn_name2id) op_name = name_generator("embedding", self.nn_name2id)
output_name = node.name output_name = node.name
layer_outputs = [op_name, output_name] layer_outputs = [op_name, output_name]
self.weights['.weight'] = _const_weight_or_none(val_x) self.weights[op_name + '.weight'] = _const_weight_or_none(val_x)
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
'paddle.nn.Embedding', 'paddle.nn.Embedding',
inputs={"x": indices_cast}, inputs={"x": indices_cast},
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或