refactor get_backbone and a little change

2021-12-22 22:16:56 +08:00
parent b74d6f7959
commit 51bb8b72b4
5 changed files with 21 additions and 26 deletions
@@ -88,7 +88,7 @@ class GaitPart(BaseModel):

    def build_network(self, model_cfg):

-        self.Backbone = self.get_backbone(model_cfg)
+        self.Backbone = self.get_backbone(model_cfg['backbone_cfg'])
        head_cfg = model_cfg['SeparateFCs']
        self.Head = SeparateFCs(**model_cfg['SeparateFCs'])
        self.Backbone = SetBlockWrapper(self.Backbone)