From 070f12804050ee054979e44e77429783f677af9a Mon Sep 17 00:00:00 2001 From: dongshuilong Date: Sat, 9 Oct 2021 08:42:12 +0000 Subject: [PATCH] fix distributied training bug for rec slim model --- ppcls/configs/Vehicle/ResNet50.yaml | 4 ++-- ppcls/configs/slim/ResNet50_vehicle_cls_prune.yaml | 4 ++-- ppcls/configs/slim/ResNet50_vehicle_cls_quantization.yaml | 6 +++--- ppcls/configs/slim/ResNet50_vehicle_reid_prune.yaml | 4 ++-- ppcls/configs/slim/ResNet50_vehicle_reid_quantization.yaml | 4 ++-- 5 files changed, 11 insertions(+), 11 deletions(-) diff --git a/ppcls/configs/Vehicle/ResNet50.yaml b/ppcls/configs/Vehicle/ResNet50.yaml index 6994789d..ba900894 100644 --- a/ppcls/configs/Vehicle/ResNet50.yaml +++ b/ppcls/configs/Vehicle/ResNet50.yaml @@ -87,9 +87,9 @@ DataLoader: mean: [0., 0., 0.] sampler: - name: DistributedRandomIdentitySampler + name: PKSampler batch_size: 128 - num_instances: 2 + sample_per_id: 2 drop_last: False shuffle: True loader: diff --git a/ppcls/configs/slim/ResNet50_vehicle_cls_prune.yaml b/ppcls/configs/slim/ResNet50_vehicle_cls_prune.yaml index 5e59e1b6..4d4f08da 100644 --- a/ppcls/configs/slim/ResNet50_vehicle_cls_prune.yaml +++ b/ppcls/configs/slim/ResNet50_vehicle_cls_prune.yaml @@ -92,9 +92,9 @@ DataLoader: mean: [0., 0., 0.] sampler: - name: DistributedRandomIdentitySampler + name: PKSampler batch_size: 128 - num_instances: 2 + sample_per_id: 2 drop_last: False shuffle: True loader: diff --git a/ppcls/configs/slim/ResNet50_vehicle_cls_quantization.yaml b/ppcls/configs/slim/ResNet50_vehicle_cls_quantization.yaml index 1ec73b0c..0e45a5a9 100644 --- a/ppcls/configs/slim/ResNet50_vehicle_cls_quantization.yaml +++ b/ppcls/configs/slim/ResNet50_vehicle_cls_quantization.yaml @@ -91,9 +91,9 @@ DataLoader: mean: [0., 0., 0.] sampler: - name: DistributedRandomIdentitySampler - batch_size: 128 - num_instances: 2 + name: PKSampler + batch_size: 64 + sample_per_id: 2 drop_last: False shuffle: True loader: diff --git a/ppcls/configs/slim/ResNet50_vehicle_reid_prune.yaml b/ppcls/configs/slim/ResNet50_vehicle_reid_prune.yaml index f9c86e2a..736c9847 100644 --- a/ppcls/configs/slim/ResNet50_vehicle_reid_prune.yaml +++ b/ppcls/configs/slim/ResNet50_vehicle_reid_prune.yaml @@ -95,9 +95,9 @@ DataLoader: mean: [0., 0., 0.] sampler: - name: DistributedRandomIdentitySampler + name: PKSampler batch_size: 128 - num_instances: 2 + sample_per_id: 2 drop_last: False shuffle: True loader: diff --git a/ppcls/configs/slim/ResNet50_vehicle_reid_quantization.yaml b/ppcls/configs/slim/ResNet50_vehicle_reid_quantization.yaml index aff5228c..72dc3186 100644 --- a/ppcls/configs/slim/ResNet50_vehicle_reid_quantization.yaml +++ b/ppcls/configs/slim/ResNet50_vehicle_reid_quantization.yaml @@ -94,9 +94,9 @@ DataLoader: mean: [0., 0., 0.] sampler: - name: DistributedRandomIdentitySampler + name: PKSampler batch_size: 64 - num_instances: 2 + sample_per_id: 2 drop_last: False shuffle: True loader: -- GitLab