fix gaitedge config and add check

for num of transform
2022-11-12 16:52:58 +08:00
parent a71444b967
commit 1588fde52d
4 changed files with 8 additions and 6 deletions
@@ -70,8 +70,8 @@ trainer_cfg:
  scheduler_reset: true
  sync_BN: true
  restore_hint:
-    - /home/leeeung/workspace/OpenGait/output/CASIA-B_new/Segmentation/Segmentation/checkpoints/Segmentation-25000.pt
-    - /home/leeeung/OpenGait/output/CASIA-B_new/GaitGL/GaitGL/checkpoints/GaitGL-80000.pt
+    - Segmentation-25000.pt
+    - GaitGL-80000.pt
  save_iter: 2000
  save_name: GaitGL_E2E
  total_iter: 20000
@@ -70,8 +70,8 @@ trainer_cfg:
  optimizer_reset: true
  scheduler_reset: true
  sync_BN: true
-  restore_hint:
-    - Segmentation-30000.pt
+  restore_hint: 
+    - Segmentation-25000.pt
    - GaitGL-80000.pt
  save_iter: 2000
  save_name: GaitEdge