From b74d6f7959586d3f12f50a00a06b8385e51b0b3c Mon Sep 17 00:00:00 2001
From: darkliang <11710911@mail.sustech.edu.cn>
Date: Tue, 21 Dec 2021 23:08:36 +0800
Subject: [PATCH] fix test bug for GLN

---
 config/gln/gln_phase1.yaml | 2 +-
 lib/modeling/models/gln.py | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/config/gln/gln_phase1.yaml b/config/gln/gln_phase1.yaml
index 7a20180..3e3c37e 100644
--- a/config/gln/gln_phase1.yaml
+++ b/config/gln/gln_phase1.yaml
@@ -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
diff --git a/lib/modeling/models/gln.py b/lib/modeling/models/gln.py
index d50a262..8e660a0 100644
--- a/lib/modeling/models/gln.py
+++ b/lib/modeling/models/gln.py
@@ -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: