Improve support for skeleton methods (#159)
* Update gaitgraph1_phase1_OUMVLP17.yaml * Update README.md
This commit is contained in:
@@ -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:
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user