From 3b17247d99725c2a12cc1df09691dd5121761a7a Mon Sep 17 00:00:00 2001 From: lvmingfu <630944715@qq.com> Date: Tue, 4 Aug 2020 18:00:20 +0800 Subject: [PATCH] fix code in debugging_in_pynative_mode.ipynb --- 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 9b523f43fa..d6c6f81bf4 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