diff --git a/research/cv/FaceAttribute/train.py b/research/cv/FaceAttribute/train.py index dbf0cf4e378b244f853b7701e7ff144929a31b7a..2fc9d7159a9b53cb50fdab3c38336630cc18d2a4 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