From 7c8559cb7c381e625f20d12ce6f59b680b7eacbd 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 eb4526425a..2582cfc635 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