diff --git a/contrib/infer.py b/contrib/infer.py
index 8f939c8455cd3868120781a7a8d96ace0ff772b1..afeb8be32f4972d139c0de4b233e1b0c859fcc4f 100644
--- a/contrib/infer.py
+++ b/contrib/infer.py
@@ -121,7 +121,7 @@ def infer():
         if (idx + 1) % 100 == 0:
             print('%d  processd' % (idx + 1))
             
-    print('%d  processd done' % (idx + 1))   
+    print('%d  processd done' % (data_num))   
     
     return 0