--- a +++ b/configs/gaitedge/phase1_rec.yaml @@ -0,0 +1,70 @@ +# 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 + data_in_use: [true, false, false, false] + dataset_partition: ./datasets/CASIA-B/CASIA-B.json + num_workers: 1 + remove_no_gallery: false + test_dataset_name: CASIA-B + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: 80000 + save_name: GaitGL + sampler: + batch_size: 4 + sample_type: all_ordered + type: InferenceSampler + transform: + - type: BaseSilTransform +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 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_float16: true + with_test: true + 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 + transform: + - type: BaseSilTransform