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
@@ -50,7 +50,7 @@ class GaitEdge(GaitGL):
        self.is_edge = model_cfg['edge']
        self.seg_lr = model_cfg['seg_lr']
        self.kernel = torch.ones(
-            (model_cfg['kernel_size'], model_cfg['kernel_size'])).cuda()
+            (model_cfg['kernel_size'], model_cfg['kernel_size']))

    def finetune_parameters(self):
        fine_tune_params = list()