diff --git a/tutorials/notebook/debugging_in_pynative_mode.ipynb b/tutorials/notebook/debugging_in_pynative_mode.ipynb index eed84a617b2f3f205640aacfa11655c39de2e66f..f0d9df64f06bf0d8cea1d4bb513220b02fbf58ad 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",