From 2b4f652c820120027a2d6e0eebc21cbc7a08092b Mon Sep 17 00:00:00 2001 From: mnhdxnh Date: Wed, 30 Jul 2025 15:59:36 +0800 Subject: [PATCH] Update 06.data_dump_MindSpore.md --- debug/accuracy_tools/msprobe/docs/06.data_dump_MindSpore.md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/debug/accuracy_tools/msprobe/docs/06.data_dump_MindSpore.md b/debug/accuracy_tools/msprobe/docs/06.data_dump_MindSpore.md index aaf7c8a99..838bd130a 100644 --- a/debug/accuracy_tools/msprobe/docs/06.data_dump_MindSpore.md +++ b/debug/accuracy_tools/msprobe/docs/06.data_dump_MindSpore.md @@ -376,8 +376,11 @@ trainer.train(1, train_dataset, callbacks=[MsprobeStep(debugger)]) ```python import mindspore as ms +from mindspore import set_device from mindspore.train import Model -ms.set_context(mode=ms.PYNATIVE_MODE, device_target="Ascend") +ms.set_context(mode=ms.PYNATIVE_MODE) + +set_device("Ascend", 0) from msprobe.mindspore import PrecisionDebugger from msprobe.mindspore.common.utils import MsprobeStep -- Gitee