Update gaitbase_hid.yaml
This commit is contained in:
@@ -1,22 +1,22 @@
|
|||||||
data_cfg:
|
data_cfg:
|
||||||
dataset_name: HID
|
dataset_name: HID
|
||||||
dataset_root: your_path
|
dataset_root: your_path
|
||||||
dataset_partition: ./datasets/HID/HID2023.json
|
dataset_partition: ./datasets/HID/HID.json
|
||||||
num_workers: 1
|
num_workers: 1
|
||||||
remove_no_gallery: false # Remove probe if no gallery for it
|
remove_no_gallery: false # Remove probe if no gallery for it
|
||||||
evaluator_cfg:
|
evaluator_cfg:
|
||||||
enable_float16: true
|
enable_float16: true
|
||||||
restore_ckpt_strict: true
|
restore_ckpt_strict: true
|
||||||
restore_hint: 60000
|
restore_hint: 60000
|
||||||
save_name: GaitBase
|
save_name: Baseline
|
||||||
eval_func: HID_submission
|
eval_func: HID_submission
|
||||||
rerank: False
|
|
||||||
sampler:
|
sampler:
|
||||||
batch_shuffle: false
|
batch_shuffle: false
|
||||||
batch_size: 8
|
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
|
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
|
frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory
|
||||||
metric: euc # cos
|
metric: euc # cos
|
||||||
|
rerank: false
|
||||||
transform:
|
transform:
|
||||||
- type: BaseSilCuttingTransform
|
- type: BaseSilCuttingTransform
|
||||||
img_w: 128
|
img_w: 128
|
||||||
@@ -31,7 +31,6 @@ loss_cfg:
|
|||||||
log_prefix: softmax
|
log_prefix: softmax
|
||||||
log_accuracy: true
|
log_accuracy: true
|
||||||
|
|
||||||
|
|
||||||
model_cfg:
|
model_cfg:
|
||||||
model: Baseline
|
model: Baseline
|
||||||
backbone_cfg:
|
backbone_cfg:
|
||||||
@@ -64,6 +63,7 @@ model_cfg:
|
|||||||
bin_num:
|
bin_num:
|
||||||
- 16
|
- 16
|
||||||
|
|
||||||
|
|
||||||
optimizer_cfg:
|
optimizer_cfg:
|
||||||
lr: 0.1
|
lr: 0.1
|
||||||
momentum: 0.9
|
momentum: 0.9
|
||||||
@@ -83,7 +83,7 @@ trainer_cfg:
|
|||||||
restore_ckpt_strict: true
|
restore_ckpt_strict: true
|
||||||
restore_hint: 0
|
restore_hint: 0
|
||||||
save_iter: 10000
|
save_iter: 10000
|
||||||
save_name: GaitBase
|
save_name: Baseline
|
||||||
sync_BN: true
|
sync_BN: true
|
||||||
total_iter: 60000
|
total_iter: 60000
|
||||||
sampler:
|
sampler:
|
||||||
|
|||||||
Reference in New Issue
Block a user