mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix distributied training bug for rec slim model

pull/1290/head
dongshuilong 2021-10-09 08:42:12 +00:00
parent 0725a66676
commit 070f128040
5 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppcls/configs/Vehicle/ResNet50.yaml
View File

@ -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:

4
ppcls/configs/slim/ResNet50_vehicle_cls_prune.yaml
View File

@ -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:

6
ppcls/configs/slim/ResNet50_vehicle_cls_quantization.yaml
View File

@ -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:

4
ppcls/configs/slim/ResNet50_vehicle_reid_prune.yaml
View File

@ -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:

4
ppcls/configs/slim/ResNet50_vehicle_reid_quantization.yaml
View File

@ -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: