提交 ac9c0ead 编写于 作者: C Channingss

modify embeding weight

上级 17cf4a57
......@@ -721,7 +721,7 @@ class OpSet9():
op_name = name_generator("embedding", self.nn_name2id)
output_name = node.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(
'paddle.nn.Embedding',
inputs={"x": indices_cast},
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册