diff --git a/ppcls/engine/evaluation/retrieval.py b/ppcls/engine/evaluation/retrieval.py
index 73fdcb3e81a590e10ef9adb4c9a86a55c80467f7..ab18950b5b774282c5e1c79ddf707f516c48e149 100644
--- 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"]