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

change sampler

上级 08582fe6
......@@ -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:
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册