Merge pull request #1289 from RainFrost1/rec_slim

fix distributied training bug for rec slim model

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:

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:

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:

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:

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: