提交 bf684b77 编写于 作者: D dengkaipeng

fix predict

上级 c4289fd7
......@@ -1622,7 +1622,7 @@ class Model(fluid.dygraph.Layer):
for k, v in zip(self._metrics_name(), metrics):
logs[k] = v
else:
outs = getattr(self, mode + '_batch')(data)
outs = getattr(self, mode + '_batch')(data[:len(self._inputs)])
outputs.append(outs)
logs['step'] = step
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册