From db53318861b93bf0c2e70536efbdafd762ba69a9 Mon Sep 17 00:00:00 2001
From: shippingwang <shipeng1108@163.com>
Date: Sat, 11 Jul 2020 10:17:49 +0000
Subject: [PATCH] refine

---
 tools/dali.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/dali.py b/tools/dali.py
index 48adf4d5..668c3a07 100644
--- a/tools/dali.py
+++ b/tools/dali.py
@@ -206,7 +206,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')
-- 
GitLab