debug

2023-03-10 08:01:20 +00:00 · 2023-03-10 08:01:20 +00:00 · 2a61ac65bf
parent ac27cb1917
commit 2a61ac65bf
1 changed files with 1 additions and 1 deletions
--- a/ppcls/engine/evaluation/retrieval.py
+++ b/ppcls/engine/evaluation/retrieval.py
@ -162,7 +162,7 @@ class RetrievalEval(object):
                has_camera = True
                batch[2] = batch[2].reshape([-1, 1]).astype("int64")

-            out = self.model(batch)
+            out = self.model(batch[0])

            if "Student" in out:
                out = out["Student"]