From 36af0f4eb1f7dc9305a9b877c8fa5ab3c565bcd7 Mon Sep 17 00:00:00 2001 From: ms_yan <6576637+ms_yan@user.noreply.gitee.com> Date: Sat, 19 Dec 2020 12:50:04 +0800 Subject: [PATCH] change num_parallel_worker config in map function --- model_zoo/official/cv/googlenet/src/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model_zoo/official/cv/googlenet/src/dataset.py b/model_zoo/official/cv/googlenet/src/dataset.py index 9aab308eb0..e95170b923 100644 --- a/model_zoo/official/cv/googlenet/src/dataset.py +++ b/model_zoo/official/cv/googlenet/src/dataset.py @@ -115,8 +115,8 @@ def create_dataset_imagenet(dataset_path, repeat_num=1, training=True, transform_label = [C.TypeCast(mstype.int32)] - data_set = data_set.map(input_columns="image", num_parallel_workers=8, operations=transform_img) - data_set = data_set.map(input_columns="label", num_parallel_workers=8, operations=transform_label) + data_set = data_set.map(input_columns="image", num_parallel_workers=12, operations=transform_img) + data_set = data_set.map(input_columns="label", num_parallel_workers=4, operations=transform_label) # apply batch operations data_set = data_set.batch(imagenet_cfg.batch_size, drop_remainder=True)