fix test bug for GLN

2021-12-21 23:08:36 +08:00
parent c3caba81c8
commit b74d6f7959
2 changed files with 1 additions and 2 deletions
@@ -14,7 +14,7 @@ evaluator_cfg:
  restore_hint: 40000
  save_name: GLN_P1
  sampler:
-    batch_size: 2
+    batch_size: 8
    sample_type: all_ordered
    type: InferenceSampler
  metric: euc # euc
@@ -86,7 +86,6 @@ class GLN(BaseModel):

    def forward(self, inputs):
        ipts, labs, _, _, seqL = inputs
-        seqL = None if not self.training else seqL
        sils = ipts[0]  # [n, s, h, w]
        del ipts
        if len(sils.size()) == 4: