From 9aec9d40745229b5b2f42aa4395515edfda85164 Mon Sep 17 00:00:00 2001 From: lvmingfu <630944715@qq.com> Date: Tue, 4 Aug 2020 18:09:41 +0800 Subject: [PATCH] fix code in debugging_in_pynative_mode.ipynb for r0.6 --- tutorials/notebook/debugging_in_pynative_mode.ipynb | 1 + 1 file changed, 1 insertion(+) diff --git a/tutorials/notebook/debugging_in_pynative_mode.ipynb b/tutorials/notebook/debugging_in_pynative_mode.ipynb index eed84a617b..f0d9df64f0 100644 --- a/tutorials/notebook/debugging_in_pynative_mode.ipynb +++ b/tutorials/notebook/debugging_in_pynative_mode.ipynb @@ -362,6 +362,7 @@ "outputs": [], "source": [ "from mindspore import context, Tensor, ParameterTuple\n", + "context.set_context(mode=context.PYNATIVE_MODE, device_target = \"GPU\")\n", "\n", "class GradWrap(nn.Cell):\n", " \"\"\" GradWrap definition \"\"\"\n", -- Gitee