diff --git a/PaddleCV/image_classification/dali.py b/PaddleCV/image_classification/dali.py
index f7c88b10ad9471b5673cb11fa9c90aa690d6ec7f..df45564c994b13b3c3ba0531bb82ae8287794fb2 100644
--- a/PaddleCV/image_classification/dali.py
+++ b/PaddleCV/image_classification/dali.py
@@ -194,7 +194,7 @@ def build(settings, mode='train'):
             pipe, ['feed_image', 'feed_label'],
             size=len(pipe),
             dynamic_shape=True,
-            fill_last_batch=False,
+            fill_last_batch=True,
             last_batch_padded=True)
 
     file_list = os.path.join(file_root, 'train_list.txt')