Improve support for skeleton methods (#159)

* Update gaitgraph1_phase1_OUMVLP17.yaml
* Update README.md
This commit is contained in:
Dongyang Jin
2023-10-17 14:31:50 +08:00
committed by GitHub
parent 2701e4b700
commit 453cc247b4
5 changed files with 63 additions and 8 deletions
@@ -11,7 +11,7 @@ evaluator_cfg:
enable_float16: false
restore_ckpt_strict: true
restore_hint: 150000
save_name: GaitGraph1_phase1
save_name: GaitGraph1_phase1_17
sampler:
batch_size: 256
frames_num_fixed: 25
@@ -26,7 +26,7 @@ evaluator_cfg:
trf_cfg:
- type: SelectSequenceCenter
sequence_length: 17
- type: SkeletonInput
- type: GaitGraph1Input
loss_cfg:
- loss_term_weight: 1
@@ -71,7 +71,7 @@ trainer_cfg:
restore_ckpt_strict: false
restore_hint: 0
save_iter: 10000
save_name: GaitGraph1_phase1
save_name: GaitGraph1_phase1_17
sync_BN: true
total_iter: 150000
sampler:
+2 -2
View File
@@ -10,7 +10,7 @@ evaluator_cfg:
enable_float16: false
restore_ckpt_strict: true
restore_hint: 80000
save_name: GaitGraph2_phase2
save_name: GaitGraph2_phase2_17
sampler:
batch_size: 256
frames_num_fixed: 30
@@ -74,7 +74,7 @@ trainer_cfg:
restore_ckpt_strict: false
restore_hint: 0
save_iter: 10000
save_name: GaitGraph2
save_name: GaitGraph2_phase2_17
sync_BN: true
total_iter: 80000
sampler:
+2 -2
View File
@@ -11,7 +11,7 @@ evaluator_cfg:
enable_float16: false
restore_ckpt_strict: true
restore_hint: 150000
save_name: GaitTR
save_name: GaitTR_17
eval_func: evaluate_indoor_dataset
sampler:
batch_size: 4 #should same to num_gpus
@@ -63,7 +63,7 @@ trainer_cfg:
restore_ckpt_strict: false
restore_hint: 0
save_iter: 5000
save_name: GaitTR
save_name: GaitTR_17
sync_BN: true
total_iter: 150000
sampler: