From fc2bb478a3d4f97936cecc356677a98b0a24060b Mon Sep 17 00:00:00 2001 From: zhanglin Date: Sat, 17 Apr 2021 20:29:47 +0800 Subject: [PATCH] fix issue https://gitee.com/mindspore/docs/issues/I3J8XZ --- tutorials/source_zh_cn/optimization.ipynb | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tutorials/source_zh_cn/optimization.ipynb b/tutorials/source_zh_cn/optimization.ipynb index ddfbfdcd27..1c3b2ebbe9 100644 --- a/tutorials/source_zh_cn/optimization.ipynb +++ b/tutorials/source_zh_cn/optimization.ipynb @@ -111,15 +111,11 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", - "\n", - "import mindspore.ops as ops\n", "import mindspore.dataset as ds\n", "import mindspore.dataset.transforms.c_transforms as C\n", "import mindspore.dataset.vision.c_transforms as CV\n", "from mindspore import nn, Tensor, Model\n", "from mindspore import dtype as mstype\n", - "from mindspore import Parameter\n", "\n", "DATA_DIR = \"./datasets/cifar-10-batches-bin/train\"\n", "\n", @@ -152,6 +148,9 @@ " return x\n", "\n", "net = Net()\n", + "epochs = 5\n", + "batch_size = 64\n", + "learning_rate = 1e-3\n", "\n", "# 构建数据集\n", "sampler = ds.SequentialSampler(num_samples=128)\n", -- Gitee