From 24a51ae06493b95501563c5ccc170421dd56edcd Mon Sep 17 00:00:00 2001 From: JamesLim-sy Date: Tue, 4 Jan 2022 07:20:31 +0000 Subject: [PATCH] first --- ppcls/configs/ImageNet/HRNet/HRNet_W48_C.yaml | 3 +++ ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml | 2 +- ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml | 2 +- ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x1_0.yaml | 2 +- ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_0.yaml | 2 +- 5 files changed, 7 insertions(+), 4 deletions(-) diff --git a/ppcls/configs/ImageNet/HRNet/HRNet_W48_C.yaml b/ppcls/configs/ImageNet/HRNet/HRNet_W48_C.yaml index 096910472..6124fe26d 100644 --- a/ppcls/configs/ImageNet/HRNet/HRNet_W48_C.yaml +++ b/ppcls/configs/ImageNet/HRNet/HRNet_W48_C.yaml @@ -1,3 +1,6 @@ +# Conv config +export FLAGS_cudnn_exhaustive_search=True + # global configs Global: checkpoints: null diff --git a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml index 53d1d1dd2..401eaa944 100644 --- a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml +++ b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml @@ -71,7 +71,7 @@ DataLoader: drop_last: False shuffle: True loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Eval: diff --git a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml index fc0e61bb6..143885d26 100644 --- a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml +++ b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml @@ -69,7 +69,7 @@ DataLoader: drop_last: False shuffle: True loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Eval: diff --git a/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x1_0.yaml b/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x1_0.yaml index 56544d0f3..f3ca83870 100644 --- a/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x1_0.yaml +++ b/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x1_0.yaml @@ -70,7 +70,7 @@ DataLoader: drop_last: False shuffle: True loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Eval: diff --git a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_0.yaml b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_0.yaml index dd4820da4..22c8d63a0 100644 --- a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_0.yaml +++ b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_0.yaml @@ -68,7 +68,7 @@ DataLoader: drop_last: False shuffle: True loader: - num_workers: 4 + num_workers: 8 use_shared_memory: True Eval: