From 97a8bb7fda53c358365ea8cf2fc580bed6752f04 Mon Sep 17 00:00:00 2001
From: Tingquan Gao <35441050@qq.com>
Date: Tue, 14 Mar 2023 16:16:40 +0800
Subject: [PATCH] Revert "debug"

This reverts commit 2a61ac65bf5d3853c3fb9a27669e265c1b1c7c01.
---
 ppcls/engine/evaluation/retrieval.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppcls/engine/evaluation/retrieval.py b/ppcls/engine/evaluation/retrieval.py
index 73fdcb3e..ab18950b 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"]
-- 
GitLab