From 1bf7264771fb30b588c426e9c153f9b643256971 Mon Sep 17 00:00:00 2001
From: Zhen Wang <wangzhen31@baidu.com>
Date: Tue, 19 May 2020 17:07:47 +0800
Subject: [PATCH] fix core dumped during the test stage of ResNet50 AMP.
 (#4646)

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

diff --git a/PaddleCV/image_classification/dali.py b/PaddleCV/image_classification/dali.py
index f7c88b10..df45564c 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')
-- 
GitLab