diff --git a/tutorials/source_zh_cn/beginner/save_load.ipynb b/tutorials/source_zh_cn/beginner/save_load.ipynb
index 559d246f146a6e3d64ae5b6f769bfe5754bff97a..93cb34f5527a41007f72e87984292804a104f117 100644
--- a/tutorials/source_zh_cn/beginner/save_load.ipynb
+++ b/tutorials/source_zh_cn/beginner/save_load.ipynb
@@ -27,7 +27,6 @@
    "outputs": [],
    "source": [
     "import mindspore.nn as nn\n",
-    "from mindspore import context\n",
     "from mindspore.train import Model\n",
     "\n",
     "from mindvision.classification.dataset import Mnist\n",
@@ -35,7 +34,6 @@
     "from mindvision.engine.callback import LossMonitor\n",
     "\n",
     "epochs = 1\n",
-    "context.set_context(mode=context.GRAPH_MODE)\n",
     "\n",
     "# 1. 构建数据集\n",
     "download_train = Mnist(path=\"./mnist\", split=\"train\", batch_size=32, repeat_num=1, shuffle=True, resize=32, download=True)\n",
diff --git a/tutorials/source_zh_cn/beginner/train.ipynb b/tutorials/source_zh_cn/beginner/train.ipynb
index 435393e475af91167106534c1499ac046e065021..90ec6ada0a31d7bd0a156a4e5b6475608d671688 100644
--- a/tutorials/source_zh_cn/beginner/train.ipynb
+++ b/tutorials/source_zh_cn/beginner/train.ipynb
@@ -125,15 +125,12 @@
    "outputs": [],
    "source": [
     "import mindspore.nn as nn\n",
-    "from mindspore import context\n",
     "from mindspore.train import Model\n",
     "\n",
     "from mindvision.classification.dataset import Mnist\n",
     "from mindvision.classification.models import lenet\n",
     "from mindvision.engine.callback import LossMonitor\n",
     "\n",
-    "context.set_context(mode=context.GRAPH_MODE)\n",
-    "\n",
     "# 1. 构建数据集\n",
     "download_train = Mnist(path=\"./mnist\", split=\"train\", batch_size=32, repeat_num=1, shuffle=True, resize=32, download=True)\n",
     "dataset_train = download_train.run()\n",