From 11e4781b364a12dfe5260a1c3e6a24a253dba50e Mon Sep 17 00:00:00 2001
From: yuchaojie <yuchaojie1@huawei.com>
Date: Sat, 17 Sep 2022 15:58:22 +0800
Subject: [PATCH] fix unet timeout
---
official/cv/unet/train.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/official/cv/unet/train.py b/official/cv/unet/train.py
index 22543d39a..1e015e831 100644
--- a/official/cv/unet/train.py
+++ b/official/cv/unet/train.py
@@ -129,7 +129,8 @@ def train_net(cross_valid_ind=1,
if __name__ == '__main__':
logging.basicConfig(level=logging.INFO, format='%(levelname)s: %(message)s')
- context.set_context(mode=context.GRAPH_MODE, device_target=config.device_target, save_graphs=False)
+ # to keep GetNext from timeout, set op_timeout=600
+ context.set_context(mode=context.GRAPH_MODE, device_target=config.device_target, save_graphs=False, op_timeout=600)
if config.device_target == "Ascend":
device_id = get_device_id()
context.set_context(device_id=device_id)
--
GitLab