diff --git a/tutorials/source_zh_cn/optimization.ipynb b/tutorials/source_zh_cn/optimization.ipynb index ddfbfdcd276985602aa9c26a3ecb663635ecd0a6..1c3b2ebbe902f711c88f532637706b2bdfc1cc25 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",