From 06651c480cc135a3e690b737626a5224277420da Mon Sep 17 00:00:00 2001
From: Zeng Jinle <32832641+sneaxiy@users.noreply.github.com>
Date: Fri, 6 Aug 2021 08:31:49 +0800
Subject: [PATCH] Fix ut test_pe_fix_op_run_order by using smaller model and
 batch size (#34603)

* fix ut

* decrease gpu memory consumption

* remove exclusive
---
 .../unittests/test_parallel_executor_fix_op_run_order.py    | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/paddle/fluid/tests/unittests/test_parallel_executor_fix_op_run_order.py b/python/paddle/fluid/tests/unittests/test_parallel_executor_fix_op_run_order.py
index f48cfbd50e..24aa080e68 100644
--- a/python/paddle/fluid/tests/unittests/test_parallel_executor_fix_op_run_order.py
+++ b/python/paddle/fluid/tests/unittests/test_parallel_executor_fix_op_run_order.py
@@ -16,7 +16,7 @@ import paddle
 import paddle.fluid as fluid
 import unittest
 import numpy as np
-from paddle.vision.models import resnet50
+from paddle.vision.models import resnet18
 from paddle.nn import CrossEntropyLoss
 
 
@@ -33,7 +33,7 @@ class TestFixOpRunOrder(unittest.TestCase):
         ) else paddle.CPUPlace()
 
     def get_feed(self):
-        batch_size = 32
+        batch_size = 4
         image = np.random.random([batch_size, 3, 224, 224]).astype('float32')
         label = np.random.randint(0, 1000, [batch_size, 1]).astype('int64')
         return {"image": image, "label": label}
@@ -47,7 +47,7 @@ class TestFixOpRunOrder(unittest.TestCase):
                 name="image", shape=[None, 3, 224, 224], dtype="float32")
             label = paddle.static.data(
                 name="label", shape=[None, 1], dtype="int64")
-            model = resnet50()
+            model = resnet18()
             pred = model(image)
             loss_fn = CrossEntropyLoss()
             loss = loss_fn(pred, label)
-- 
GitLab