From b50947d2e2c639a7c4891bce20a5daaef888bfae Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 14 Sep 2022 15:26:23 +0800
Subject: [PATCH] docs(mge/module): add Module.state_dict docstring

GitOrigin-RevId: 424ec791d7eda63d2835154d3d2e985a8915f7d5
---
 imperative/python/megengine/module/module.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/imperative/python/megengine/module/module.py b/imperative/python/megengine/module/module.py
index 8dca6dae8..42625d760 100644
--- a/imperative/python/megengine/module/module.py
+++ b/imperative/python/megengine/module/module.py
@@ -452,6 +452,7 @@ class Module(metaclass=ABCMeta):
         return offset
 
     def state_dict(self, rst=None, prefix="", keep_var=False):
+        r"""Returns a dictionary containing whole states of the module."""
         _rst = self._state_dict(rst=rst, prefix=prefix, keep_var=keep_var)
         rst = OrderedDict()
         XNorm_typeclass = _get_XNorm_typeclass()
-- 
GitLab