Add Group Normalization

5 years ago · 898acc3201
parent ebe6efff71
commit 898acc3201
1 changed files with 7 additions and 5 deletions
--- a/mindspore/nn/layer/normalization.py
+++ b/mindspore/nn/layer/normalization.py
@ -357,4 +357,6 @@ class GroupNorm(Cell):
        return output

    def extend_repr(self):
-        return 'num_groups={}, num_channels={}'.format(self.num_groups,self.num_channels)
+        """Display instance object as string."""
+        s = 'num_groups={}, num_channels={}'.format(self.num_groups, self.num_channels)
+        return s