未验证 提交 74f19d78 编写于 作者: K Kaipeng Deng 提交者: GitHub

Merge pull request #76 from heavengate/fix_predict

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