Update gaitbase_hid.yaml

configs/gaitbase/gaitbase_hid.yaml

@@ -34,35 +34,34 @@ loss_cfg:
 model_cfg:
   model: Baseline
   backbone_cfg:
-    type: ResNet9
+    in_channels: 1
-    block: BasicBlock
+    layers_cfg: # Layers configuration for automatically model construction
-    channels: # Layers configuration for automatically model construction
+      - BC-64
-      - 64
+      - BC-64
-      - 128
+      - M
-      - 256
+      - BC-128
-      - 512
+      - BC-128
-    layers:
+      - M
-      - 1
+      - BC-256
-      - 1
+      - BC-256
-      - 1
+      - M
-      - 1
+      - BC-512
-    strides:
+      - BC-512
-      - 1
+    type: Plain
-      - 2
-      - 2
-      - 1
-    maxpool: false
   SeparateFCs:
     in_channels: 512
     out_channels: 256
-    parts_num: 16
+    parts_num: 31
   SeparateBNNecks:
-    class_num: 500 # you need to change
+    class_num: 500
     in_channels: 256
-    parts_num: 16
+    parts_num: 31
   bin_num:
     - 16
-
+    - 8
+    - 4
+    - 2
+    - 1
 
 optimizer_cfg:
   lr: 0.1