change sampler

pull/1819/head
weishengyu 2022-04-06 20:14:29 +08:00
parent 08582fe617
commit 5615bc6cfd
2 changed files with 4 additions and 4 deletions

View File

@ -4,7 +4,7 @@ Global:
pretrained_model: null
output_dir: "./output/"
device: "gpu"
save_interval: 1
save_interval: 10
eval_during_train: True
eval_interval: 10
epochs: 120
@ -85,9 +85,9 @@ DataLoader:
r1: 0.3
mean: [0.4914, 0.4822, 0.4465]
sampler:
name: PKSampler
name: DistributedRandomIdentitySampler
batch_size: 64
sample_per_id: 4
num_instances: 4
drop_last: True
shuffle: True
loader:

View File

@ -4,7 +4,7 @@ Global:
pretrained_model: null
output_dir: "./output/"
device: "gpu"
save_interval: 1
save_interval: 10
eval_during_train: True
eval_interval: 10
epochs: 120