From 2e8742086563ea442c357b14560245c54e0aa0a3 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Sat, 21 Nov 2020 17:32:08 +0800 Subject: [PATCH] fix(module): fix docs in normalization GitOrigin-RevId: 326234c35a9c985c2cf75e3b1ee7953237495fb5 --- imperative/python/megengine/module/normalization.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/imperative/python/megengine/module/normalization.py b/imperative/python/megengine/module/normalization.py index bdd90f7a1..8b28eba85 100644 --- a/imperative/python/megengine/module/normalization.py +++ b/imperative/python/megengine/module/normalization.py @@ -14,7 +14,7 @@ from megengine import Parameter class GroupNorm(M.Module): """ - Simple implementation of GroupNorm. + Simple implementation of GroupNorm. Only support 4d tensor now. Reference: https://arxiv.org/pdf/1803.08494.pdf. """ @@ -63,7 +63,7 @@ class GroupNorm(M.Module): class InstanceNorm(M.Module): """ - simple implementation of InstanceNorm. + Simple implementation of InstanceNorm. Only support 4d tensor now. Reference: https://arxiv.org/abs/1607.08022. Note that InstanceNorm equals using GroupNome with num_groups=num_channels. """ @@ -107,7 +107,7 @@ class InstanceNorm(M.Module): class LayerNorm(M.Module): """ - simple implementation of LayerNorm. + Simple implementation of LayerNorm. Only support 4d tensor now. Reference: https://arxiv.org/pdf/1803.08494.pdf. Note that LayerNorm equals using GroupNorm with num_groups=1. """ -- GitLab