提交 5615bc6c 编写于 作者: W weishengyu

change sampler

上级 08582fe6
...@@ -4,7 +4,7 @@ Global: ...@@ -4,7 +4,7 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: "./output/" output_dir: "./output/"
device: "gpu" device: "gpu"
save_interval: 1 save_interval: 10
eval_during_train: True eval_during_train: True
eval_interval: 10 eval_interval: 10
epochs: 120 epochs: 120
...@@ -85,9 +85,9 @@ DataLoader: ...@@ -85,9 +85,9 @@ DataLoader:
r1: 0.3 r1: 0.3
mean: [0.4914, 0.4822, 0.4465] mean: [0.4914, 0.4822, 0.4465]
sampler: sampler:
name: PKSampler name: DistributedRandomIdentitySampler
batch_size: 64 batch_size: 64
sample_per_id: 4 num_instances: 4
drop_last: True drop_last: True
shuffle: True shuffle: True
loader: loader:
......
...@@ -4,7 +4,7 @@ Global: ...@@ -4,7 +4,7 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: "./output/" output_dir: "./output/"
device: "gpu" device: "gpu"
save_interval: 1 save_interval: 10
eval_during_train: True eval_during_train: True
eval_interval: 10 eval_interval: 10
epochs: 120 epochs: 120
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册