From fdb7bbf422c3406762d0ca39499b3df13f5e23f2 Mon Sep 17 00:00:00 2001 From: mwang Date: Mon, 22 Mar 2021 17:52:17 +0800 Subject: [PATCH] fix thor --- tests/st/networks/models/resnet50/src_thor/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/st/networks/models/resnet50/src_thor/dataset.py b/tests/st/networks/models/resnet50/src_thor/dataset.py index 69524c42b19..83efe83e3a8 100644 --- a/tests/st/networks/models/resnet50/src_thor/dataset.py +++ b/tests/st/networks/models/resnet50/src_thor/dataset.py @@ -72,8 +72,8 @@ def create_dataset(dataset_path, do_train, repeat_num=1, batch_size=32): type_cast_op = C2.TypeCast(mstype.int32) - data_set = data_set.map(operations=trans, input_columns="image", num_parallel_workers=8) - data_set = data_set.map(operations=type_cast_op, input_columns="label", num_parallel_workers=8) + data_set = data_set.map(operations=trans, input_columns="image", num_parallel_workers=24) + data_set = data_set.map(operations=type_cast_op, input_columns="label", num_parallel_workers=24) # apply batch operations data_set = data_set.batch(batch_size, drop_remainder=True)