update gaitedge and use data_in_use

This commit is contained in:
darkliang
2022-07-19 17:34:20 +08:00
parent 4b681fb9bd
commit 8edfc80326
7 changed files with 66 additions and 24 deletions
+3 -2
View File
@@ -2,13 +2,14 @@
data_cfg:
dataset_name: CASIA-B*
dataset_root: your_path
data_in_use: [true, false, false, false]
dataset_partition: ./datasets/CASIA-B*/CASIA-B*.json
num_workers: 1
remove_no_gallery: false
test_dataset_name: CASIA-B
evaluator_cfg:
enable_float16: true
enable_float16: true
restore_ckpt_strict: true
restore_hint: 80000
save_name: GaitGL
@@ -66,4 +67,4 @@ trainer_cfg:
sample_type: fixed_ordered
type: TripletSampler
transform:
- type: BaseSilTransform
- type: BaseSilTransform