From e0de28776ce7f7bd5f3931e98a887989db1e6540 Mon Sep 17 00:00:00 2001 From: wukesong Date: Fri, 15 May 2020 11:56:00 +0800 Subject: [PATCH] add look_sink=fasle for CPU&GPU --- tutorials/source_en/quick_start/quick_start.md | 1 + tutorials/source_zh_cn/quick_start/quick_start.md | 1 + tutorials/tutorial_code/lenet.py | 1 + 3 files changed, 3 insertions(+) diff --git a/tutorials/source_en/quick_start/quick_start.md b/tutorials/source_en/quick_start/quick_start.md index c14baceec3..f2663c28cd 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 605238850a..5ce9c9b646 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 797c63b9da..db2e99fbba 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 -- Gitee