From bf684b77e035f4a0cf185ee188a18ec61d63ae63 Mon Sep 17 00:00:00 2001
From: dengkaipeng <dengkaipeng@baidu.com>
Date: Sat, 9 May 2020 03:38:52 +0000
Subject: [PATCH] fix predict

---
 hapi/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hapi/model.py b/hapi/model.py
index 083d2ee..9bb307a 100644
--- a/hapi/model.py
+++ b/hapi/model.py
@@ -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
-- 
GitLab