Update gaitbase_hid.yaml
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user