mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

debug

pull/2698/head
gaotingquan 2023-03-10 08:01:20 +00:00 committed by Wei Shengyu
parent ac27cb1917
commit 2a61ac65bf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/engine/evaluation/retrieval.py
View File

@ -162,7 +162,7 @@ class RetrievalEval(object):
has_camera = True has_camera = True
batch[2] = batch[2].reshape([-1, 1]).astype("int64") batch[2] = batch[2].reshape([-1, 1]).astype("int64")
out = self.model(batch) out = self.model(batch[0])
if "Student" in out: if "Student" in out:
out = out["Student"] out = out["Student"]