crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix test bug for GLN

Browse Source
This commit is contained in:
darkliang
2021-12-21 23:08:36 +08:00
parent c3caba81c8
commit b74d6f7959
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
config/gln/gln_phase1.yaml
+1 -1
View File
@@ -14,7 +14,7 @@ evaluator_cfg:
restore_hint: 40000 restore_hint: 40000
save_name: GLN_P1 save_name: GLN_P1
sampler: sampler:
batch_size: 2 batch_size: 8
sample_type: all_ordered sample_type: all_ordered
type: InferenceSampler type: InferenceSampler
metric: euc # euc metric: euc # euc
lib/modeling/models/gln.py
-1
View File
@@ -86,7 +86,6 @@ class GLN(BaseModel):
def forward(self, inputs): def forward(self, inputs):
ipts, labs, _, _, seqL = inputs ipts, labs, _, _, seqL = inputs
seqL = None if not self.training else seqL
sils = ipts[0] # [n, s, h, w] sils = ipts[0] # [n, s, h, w]
del ipts del ipts
if len(sils.size()) == 4: if len(sils.size()) == 4: