Solve the problem of dimension misuse. (#59)

* commit for fix dimension

* fix dimension for all method

* restore config

* clean up baseline config

* add contiguous

* rm comment

config/baseline/baseline.yaml

@@ -46,9 +46,6 @@ model_cfg:
       - M
       - BC-256
       - BC-256
-      # - M
-      # - BC-512
-      # - BC-512
     type: Plain
   SeparateFCs:
     in_channels: 256
@@ -81,6 +78,7 @@ trainer_cfg:
   enable_float16: true # half_percesion float for memory reduction and speedup
   fix_BN: false
   log_iter: 100
+  with_test: true
   restore_ckpt_strict: true
   restore_hint: 0
   save_iter: 10000

config/gaitgl/gaitgl.yaml

@@ -47,8 +47,8 @@ scheduler_cfg:
   scheduler: MultiStepLR
 
 trainer_cfg:
-  enable_distributed: true
-  enable_float16: false
+  enable_float16: true
+  with_test: true
   log_iter: 100
   restore_ckpt_strict: true
   restore_hint: 0

config/gaitpart/gaitpart.yaml

@@ -59,6 +59,7 @@ scheduler_cfg:
 trainer_cfg:
   enable_float16: true
   log_iter: 100
+  with_test: true
   restore_ckpt_strict: true
   restore_hint: 0
   save_iter: 10000

config/gaitset/gaitset.yaml

@@ -57,12 +57,13 @@ scheduler_cfg:
 trainer_cfg:
   enable_float16: true
   log_iter: 100
+  with_test: true
   restore_ckpt_strict: true
   restore_hint: 0
   save_iter: 10000
   save_name: GaitSet
   sync_BN: false
-  total_iter: 42000
+  total_iter: 40000
   sampler:
     batch_shuffle: false
     batch_size:

config/gln/gln_phase1.yaml

@@ -74,10 +74,9 @@ scheduler_cfg:
   scheduler: MultiStepLR
 
 trainer_cfg:
-  enable_distributed: true
   enable_float16: true
   fix_layers: false
-  with_test: false
+  with_test: true
   log_iter: 100
   optimizer_reset: false
   restore_ckpt_strict: true