From 01294ff9b6f78688636b5d0422f74ced2ddf847b Mon Sep 17 00:00:00 2001
From: Huihuang Zheng <zhhsplendid@gmail.com>
Date: Thu, 21 May 2020 10:43:19 +0800
Subject: [PATCH] Decrease the batch size of se-resnext in unittest. (#24651)

As discussed with QA, we will use p4 machine for unit test and the GPU on those machine may not have enough GPU, which can cause "test_parallel_executor_seresnext_base_gpu" failed. So I decrease the batch size.
---
 python/paddle/fluid/tests/unittests/seresnext_net.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/fluid/tests/unittests/seresnext_net.py b/python/paddle/fluid/tests/unittests/seresnext_net.py
index 5559fb7eab0..c31471efbde 100644
--- a/python/paddle/fluid/tests/unittests/seresnext_net.py
+++ b/python/paddle/fluid/tests/unittests/seresnext_net.py
@@ -173,7 +173,7 @@ model = SE_ResNeXt50Small
 
 
 def batch_size():
-    return 12
+    return 8
 
 
 def iter(use_cuda):
-- 
GitLab