From 339e6c412c828d1dd4ffdfcfae133747ab91777a Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Thu, 26 Nov 2020 22:01:02 +0800
Subject: [PATCH] Initialize batch_size, prevent gpu memory overflow in infer
 (#1768)

---
 ppdet/data/reader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppdet/data/reader.py b/ppdet/data/reader.py
index 3dce62d8e..cb5549892 100644
--- a/ppdet/data/reader.py
+++ b/ppdet/data/reader.py
@@ -190,7 +190,7 @@ class Reader(object):
                  dataset=None,
                  sample_transforms=None,
                  batch_transforms=None,
-                 batch_size=None,
+                 batch_size=1,
                  shuffle=False,
                  drop_last=False,
                  drop_empty=True,
-- 
GitLab