Junhao Liang
bb6cd5149a
* fix bug in fix_BN * gaitgl OUMVLP support. * update ./doc/3.advance_usage.md Cross-Dataset Evalution & Data Agumentation * update config * update docs.3 * update docs.3 * add loss doc and gather input decorator * refine the create model doc * support rearrange directory of unzipped OUMVLP * fix some bugs in loss_aggregator.py * refine docs and little fix * add oumvlp pretreatment description * pretreatment dataset fix oumvlp description * add gaitgl oumvlp result * assert gaitgl input size * add pipeline * update the readme. * update pipeline and readme * Corrigendum. * add logo and remove path * update new logo * Update README.md * modify logo size Co-authored-by: 12131100 <12131100@mail.sustech.edu.cn> Co-authored-by: noahshen98 <77523610+noahshen98@users.noreply.github.com> Co-authored-by: Noah <595311942@qq.com>
68 lines
1.3 KiB
YAML
68 lines
1.3 KiB
YAML
# Note : *** the batch_size should be equal to the gpus number at the test phase!!! ***
|
|
data_cfg:
|
|
dataset_name: CASIA-B
|
|
dataset_root: your_path
|
|
dataset_partition: ./misc/partitions/CASIA-B_include_005.json
|
|
num_workers: 1
|
|
remove_no_gallery: false
|
|
test_dataset_name: CASIA-B
|
|
|
|
evaluator_cfg:
|
|
enable_distributed: true
|
|
enable_float16: false
|
|
restore_ckpt_strict: true
|
|
restore_hint: 80000
|
|
save_name: GaitGL
|
|
sampler:
|
|
batch_size: 4
|
|
sample_type: all_ordered
|
|
type: InferenceSampler
|
|
|
|
loss_cfg:
|
|
- loss_term_weights: 1.0
|
|
margin: 0.2
|
|
type: TripletLoss
|
|
log_prefix: triplet
|
|
- loss_term_weights: 1.0
|
|
scale: 1
|
|
type: CrossEntropyLoss
|
|
log_accuracy: true
|
|
label_smooth: false
|
|
log_prefix: softmax
|
|
|
|
model_cfg:
|
|
model: GaitGL
|
|
channels: [32, 64, 128]
|
|
class_num: 74
|
|
|
|
optimizer_cfg:
|
|
lr: 1.0e-4
|
|
solver: Adam
|
|
weight_decay: 5.0e-4
|
|
|
|
scheduler_cfg:
|
|
gamma: 0.1
|
|
milestones:
|
|
- 70000
|
|
scheduler: MultiStepLR
|
|
|
|
trainer_cfg:
|
|
enable_distributed: true
|
|
enable_float16: false
|
|
log_iter: 100
|
|
restore_ckpt_strict: true
|
|
restore_hint: 0
|
|
save_iter: 10000
|
|
save_name: GaitGL
|
|
sync_BN: true
|
|
total_iter: 80000
|
|
sampler:
|
|
batch_shuffle: true
|
|
batch_size:
|
|
- 8
|
|
- 8
|
|
frames_num_fixed: 30
|
|
frames_skip_num: 0
|
|
sample_type: fixed_ordered
|
|
type: TripletSampler
|