fix linear merge error for pangu model

2021-11-09 11:58:07 +08:00 · 2021-11-09 11:58:07 +08:00 · d1123c9b20
parent dbbe870036
commit d1123c9b20
1 changed files with 1 additions and 1 deletions
--- 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,