From 0af4680f86f4e997af368efba983697c0165b91b Mon Sep 17 00:00:00 2001 From: Yang Nie Date: Wed, 19 Apr 2023 12:19:41 +0800 Subject: [PATCH] set num_workers 8 --- ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x0_5.yaml | 4 ++-- ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_0.yaml | 4 ++-- ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_5.yaml | 4 ++-- ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x2_0.yaml | 4 ++-- 4 files changed, 8 insertions(+), 8 deletions(-) diff --git a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x0_5.yaml b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x0_5.yaml index 32fb6b93..44e43b8e 100644 --- a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x0_5.yaml +++ b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x0_5.yaml @@ -105,7 +105,7 @@ DataLoader: drop_last: False shuffle: True loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Eval: dataset: @@ -134,7 +134,7 @@ DataLoader: drop_last: False shuffle: False loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Infer: diff --git a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_0.yaml b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_0.yaml index 7e77a639..ba308d62 100644 --- a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_0.yaml +++ b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_0.yaml @@ -105,7 +105,7 @@ DataLoader: drop_last: False shuffle: True loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Eval: dataset: @@ -134,7 +134,7 @@ DataLoader: drop_last: False shuffle: False loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Infer: diff --git a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_5.yaml b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_5.yaml index 9f1df77a..79bcc80b 100644 --- a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_5.yaml +++ b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_5.yaml @@ -105,7 +105,7 @@ DataLoader: drop_last: False shuffle: True loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Eval: dataset: @@ -134,7 +134,7 @@ DataLoader: drop_last: False shuffle: False loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Infer: diff --git a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x2_0.yaml b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x2_0.yaml index 851cd64e..142c4e7d 100644 --- a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x2_0.yaml +++ b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x2_0.yaml @@ -105,7 +105,7 @@ DataLoader: drop_last: False shuffle: True loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Eval: dataset: @@ -134,7 +134,7 @@ DataLoader: drop_last: False shuffle: False loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Infer: -- GitLab