fix code in debugging_in_pynative_mode.ipynb for r0.6

9aec9d40 · lvmingfu · c9f5bd54 · 9aec9d40
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

tutorials/notebook/debugging_in_pynative_mode.ipynb tutorials/notebook/debugging_in_pynative_mode.ipynb +1 -0

未找到文件。
--- 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",