update gaitedge and use data_in_use

2022-07-19 17:34:20 +08:00
parent 4b681fb9bd
commit 8edfc80326
7 changed files with 66 additions and 24 deletions
@@ -1,6 +1,7 @@
 data_cfg:
  dataset_name: CASIA-B*
  dataset_root: your_path
+  data_in_use: [false, true, true, true]
  dataset_partition: ./datasets/CASIA-B*/CASIA-B*.json
  num_workers: 1
  remove_no_gallery: false # Remove probe if no gallery for it
@@ -68,9 +69,9 @@ trainer_cfg:
  optimizer_reset: true
  scheduler_reset: true
  sync_BN: true
-  restore_hint: 
-   - Segmentation-30000.pt
-   - GaitGL-80000.pt
+  restore_hint:
+    - Segmentation-30000.pt
+    - GaitGL-80000.pt
  save_iter: 2000
  save_name: GaitEdge
  total_iter: 20000