diff --git a/ppcls/configs/Vehicle/ResNet50.yaml b/ppcls/configs/Vehicle/ResNet50.yaml
index 6994789d5bf0e7a713d1f9b7e3d27f3fc489130c..ba9008943edfaab887f9fb6779ed79f9a77be914 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 5e59e1b6edf572f162de70698b0bdbb5f71264e1..4d4f08da50ad82f60d4bf245d19f09633a663a34 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 1ec73b0cb2c5157185cec6deef21c6f0e92c8d43..0e45a5a9bd306c58491a42b428a4971754318304 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 f9c86e2a27e01660c6cf1cd2dce7f343658cb4dc..736c9847a41b5f190eb198cb522381730e58060c 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 aff5228c12fe812941a3de350c94c7f831465daa..72dc318656939a1597f26258d44a8900fb21dec4 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: