From d1123c9b202e05862a13c2267521555811ee9db4 Mon Sep 17 00:00:00 2001 From: huangxinjing Date: Tue, 9 Nov 2021 11:58:07 +0800 Subject: [PATCH] fix linear merge error for pangu model --- mindspore/parallel/nn/layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/parallel/nn/layers.py b/mindspore/parallel/nn/layers.py index 7557b3b7337..2beb1b38bf6 100644 --- a/mindspore/parallel/nn/layers.py +++ b/mindspore/parallel/nn/layers.py @@ -319,7 +319,7 @@ class _Linear(Cell): ``Ascend`` ``GPU`` """ - @cell_attr_register(attrs=['has_bias', 'in_channels', 'out_channels', 'shard_output', 'activation']) + @cell_attr_register def __init__(self, in_channels, out_channels,