diff --git a/mindspore/nn/cell.py b/mindspore/nn/cell.py
index 42f97c039d..bd9e5692a9 100755
--- a/mindspore/nn/cell.py
+++ b/mindspore/nn/cell.py
@@ -1405,12 +1405,16 @@ class GraphCell(Cell):
         >>> from mindspore import Tensor
         >>> from mindspore.train import export, load
         >>>
-        >>> net = nn.Conv2d(1, 1, kernel_size=3)
+        >>> net = nn.Conv2d(1, 1, kernel_size=3, weight_init="ones")
         >>> input = Tensor(np.ones([1, 1, 3, 3]).astype(np.float32))
         >>> export(net, input, file_name="net", file_format="MINDIR")
         >>> graph = load("net.mindir")
         >>> net = nn.GraphCell(graph)
         >>> output = net(input)
+        >>> print(output)
+        [[[[4. 6. 4.]
+           [6. 9. 6.]
+           [4. 6. 4.]]]]
     """
     def __init__(self, graph):
         super(GraphCell, self).__init__(auto_prefix=True)
diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py
index f09cf1a349..9b018e6ef0 100644
--- a/mindspore/train/serialization.py
+++ b/mindspore/train/serialization.py
@@ -331,16 +331,16 @@ def load(file_name, **kwargs):
         >>> from mindspore import Tensor
         >>> from mindspore.train import export, load
         >>>
-        >>> net = nn.Conv2d(1, 1, kernel_size=3)
+        >>> net = nn.Conv2d(1, 1, kernel_size=3, weight_init="ones")
         >>> input = Tensor(np.ones([1, 1, 3, 3]).astype(np.float32))
         >>> export(net, input, file_name="net", file_format="MINDIR")
         >>> graph = load("net.mindir")
         >>> net = nn.GraphCell(graph)
         >>> output = net(input)
         >>> print(output)
-        [[[[0.03204346 0.04455566 0.03509521]
-           [0.02406311 0.04125977 0.02404785]
-           [0.02018738 0.0292511  0.00889587]]]]
+        [[[[4. 6. 4.]
+           [6. 9. 6.]
+           [4. 6. 4.]]]]
     """
     if not isinstance(file_name, str):
         raise ValueError("The file name must be string.")