Skip to content
Snippets Groups Projects
Unverified Commit 03da443e authored by fangzehua's avatar fangzehua Committed by Gitee
Browse files

fix FaceAttribute runtime_num_threads

parent f1e2e367
No related branches found
No related tags found
No related merge requests found
...@@ -120,7 +120,7 @@ def modelarts_pre_process(): ...@@ -120,7 +120,7 @@ def modelarts_pre_process():
def run_train(): def run_train():
'''run train.''' '''run train.'''
context.set_context(mode=context.GRAPH_MODE, device_target=config.device_target, save_graphs=False, 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) mindspore.set_seed(1)
# init distributed # init distributed
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment