From 1c405c367f3219a65b21f3b6f507131e4724a37b Mon Sep 17 00:00:00 2001
From: Dongyang Jin <73057174+jdyjjj@users.noreply.github.com>
Date: Sat, 26 Apr 2025 22:30:35 +0800
Subject: [PATCH] Update gaitbase_hid.yaml

---
 configs/gaitbase/gaitbase_hid.yaml | 43 +++++++++++++++---------------
 1 file changed, 21 insertions(+), 22 deletions(-)

diff --git a/configs/gaitbase/gaitbase_hid.yaml b/configs/gaitbase/gaitbase_hid.yaml
index 2d461ff..24085b7 100644
--- a/configs/gaitbase/gaitbase_hid.yaml
+++ b/configs/gaitbase/gaitbase_hid.yaml
@@ -34,35 +34,34 @@ loss_cfg:
 model_cfg:
   model: Baseline
   backbone_cfg:
-    type: ResNet9
-    block: BasicBlock
-    channels: # Layers configuration for automatically model construction
-      - 64
-      - 128
-      - 256
-      - 512
-    layers:
-      - 1
-      - 1
-      - 1
-      - 1
-    strides:
-      - 1
-      - 2
-      - 2
-      - 1
-    maxpool: false
+    in_channels: 1
+    layers_cfg: # Layers configuration for automatically model construction
+      - BC-64
+      - BC-64
+      - M
+      - BC-128
+      - BC-128
+      - M
+      - BC-256
+      - BC-256
+      - M
+      - BC-512
+      - BC-512
+    type: Plain
   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