From a5a9bef9fb64eb922cf1dce5418e340d12d4174f Mon Sep 17 00:00:00 2001 From: Dongyang Jin <73057174+jdyjjj@users.noreply.github.com> Date: Sat, 26 Apr 2025 22:20:54 +0800 Subject: [PATCH] Update gaitbase_hid.yaml --- configs/gaitbase/gaitbase_hid.yaml | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/configs/gaitbase/gaitbase_hid.yaml b/configs/gaitbase/gaitbase_hid.yaml index 1cc8e12..2d461ff 100644 --- a/configs/gaitbase/gaitbase_hid.yaml +++ b/configs/gaitbase/gaitbase_hid.yaml @@ -1,22 +1,22 @@ data_cfg: dataset_name: HID dataset_root: your_path - dataset_partition: ./datasets/HID/HID2023.json + dataset_partition: ./datasets/HID/HID.json num_workers: 1 remove_no_gallery: false # Remove probe if no gallery for it evaluator_cfg: enable_float16: true restore_ckpt_strict: true restore_hint: 60000 - save_name: GaitBase + save_name: Baseline eval_func: HID_submission - rerank: False sampler: batch_shuffle: false batch_size: 8 sample_type: all_ordered # all indicates whole sequence used to test, while ordered means input sequence by its natural order; Other options: fixed_unordered frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory metric: euc # cos + rerank: false transform: - type: BaseSilCuttingTransform img_w: 128 @@ -31,7 +31,6 @@ loss_cfg: log_prefix: softmax log_accuracy: true - model_cfg: model: Baseline backbone_cfg: @@ -64,6 +63,7 @@ model_cfg: bin_num: - 16 + optimizer_cfg: lr: 0.1 momentum: 0.9 @@ -83,7 +83,7 @@ trainer_cfg: restore_ckpt_strict: true restore_hint: 0 save_iter: 10000 - save_name: GaitBase + save_name: Baseline sync_BN: true total_iter: 60000 sampler: