提交 2a61ac65 编写于 作者: G gaotingquan 提交者: Wei Shengyu

debug

上级 ac27cb19
......@@ -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"]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册