diff --git a/tutorials/source_en/quick_start/quick_start.md b/tutorials/source_en/quick_start/quick_start.md index c14baceec3229dbd99a86e518a8e38b85ae83964..f2663c28cd60cf1de5377ee156960ba1d73e3aaf 100644 --- a/tutorials/source_en/quick_start/quick_start.md +++ b/tutorials/source_en/quick_start/quick_start.md @@ -101,6 +101,7 @@ if __name__ == "__main__": args = parser.parse_args() context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target, enable_mem_reuse=False) + context.set_context(enable_loop_sink=(args.device_target == 'Ascend')) ... ``` diff --git a/tutorials/source_zh_cn/quick_start/quick_start.md b/tutorials/source_zh_cn/quick_start/quick_start.md index 605238850a856f7ddb8b052e48369e87ce0717ef..5ce9c9b646245642a26fd37e18c0306d8e7f54e4 100644 --- a/tutorials/source_zh_cn/quick_start/quick_start.md +++ b/tutorials/source_zh_cn/quick_start/quick_start.md @@ -103,6 +103,7 @@ if __name__ == "__main__": args = parser.parse_args() context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target, enable_mem_reuse=False) + context.set_context(enable_loop_sink=(args.device_target == 'Ascend')) ... ``` diff --git a/tutorials/tutorial_code/lenet.py b/tutorials/tutorial_code/lenet.py index 797c63b9da9609ae5616c519e64af73692b8fdd6..db2e99fbbacb1d9adc911b62cb6acea3bf15fb81 100644 --- a/tutorials/tutorial_code/lenet.py +++ b/tutorials/tutorial_code/lenet.py @@ -197,6 +197,7 @@ if __name__ == "__main__": args = parser.parse_args() context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target, enable_mem_reuse=False) + context.set_context(enable_loop_sink=(args.device_target == 'Ascend')) # download mnist dataset download_dataset() # learning rate setting