diff --git a/ppcls/configs/Products/ResNet50_vd_Inshop.yaml b/ppcls/configs/Products/ResNet50_vd_Inshop.yaml index b29a3a3f0e53099b06cd4aa2994d3bbd209a0467..96fab34f1172f0da16d01e3db9982257377f02b0 100644 --- a/ppcls/configs/Products/ResNet50_vd_Inshop.yaml +++ b/ppcls/configs/Products/ResNet50_vd_Inshop.yaml @@ -90,10 +90,10 @@ DataLoader: r1: 0.3 mean: [0., 0., 0.] sampler: - name: DistributedRandomIdentitySampler + name: PKSampler batch_size: 64 - num_instances: 2 - drop_last: False + sample_per_id: 2 + drop_last: True shuffle: True loader: num_workers: 4