fix test bug for GLN

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
-1
View File
@@ -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: