From 03da443ea3f45e8f7c8ba519fbda9c733580a47a Mon Sep 17 00:00:00 2001
From: fangzehua <fangzehua1@huawei.com>
Date: Fri, 8 Jul 2022 01:41:32 +0000
Subject: [PATCH] fix FaceAttribute runtime_num_threads
---
research/cv/FaceAttribute/train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/research/cv/FaceAttribute/train.py b/research/cv/FaceAttribute/train.py
index dbf0cf4e3..2fc9d7159 100644
--- a/research/cv/FaceAttribute/train.py
+++ b/research/cv/FaceAttribute/train.py
@@ -120,7 +120,7 @@ def modelarts_pre_process():
def run_train():
'''run train.'''
context.set_context(mode=context.GRAPH_MODE, device_target=config.device_target, save_graphs=False,
- device_id=get_device_id())
+ runtime_num_threads=10, device_id=get_device_id())
mindspore.set_seed(1)
# init distributed
--
GitLab