diff --git a/hapi/model.py b/hapi/model.py
index 083d2ee0f8bcd38545b66b25312d4da2042140cf..9bb307a3f1de9819d3ac2c86f293254e61a14e4b 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