Revert "debug"

This reverts commit 2a61ac65.

Revert "debug"
This reverts commit 2a61ac65.
97a8bb7f · Tingquan Gao · 5fde410d · 97a8bb7f
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppcls/engine/evaluation/retrieval.py ppcls/engine/evaluation/retrieval.py +1 -1

未找到文件。
--- 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[0])
+            out = self.model(batch)

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