From 32d8e2747267b39d857c060bad46a20bb9d7ead3 Mon Sep 17 00:00:00 2001
From: x54-729 <17307130121@fudan.edu.cn>
Date: Fri, 15 Apr 2022 16:10:10 +0000
Subject: [PATCH] small

---
 tests/core/drivers/paddle_driver/test_single_device.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/core/drivers/paddle_driver/test_single_device.py b/tests/core/drivers/paddle_driver/test_single_device.py
index 12f52537..92c55434 100644
--- a/tests/core/drivers/paddle_driver/test_single_device.py
+++ b/tests/core/drivers/paddle_driver/test_single_device.py
@@ -535,7 +535,7 @@ class TestSetDistReproDataloder:
 #
 ############################################################################
 
-def generate_random_driver(features, labels, fp16, device="cpu"):
+def generate_random_driver(features, labels, fp16=False, device="cpu"):
     """
     生成driver
     """
@@ -549,8 +549,8 @@ def generate_random_driver(features, labels, fp16, device="cpu"):
 
 @pytest.fixture
 def prepare_test_save_load():
-    dataset = PaddleRandomMaxDataset(320, 10)
-    dataloader = DataLoader(dataset, batch_size=32)
+    dataset = PaddleRandomMaxDataset(40, 10)
+    dataloader = DataLoader(dataset, batch_size=4)
     driver1, driver2 = generate_random_driver(10, 10), generate_random_driver(10, 10)
     return driver1, driver2, dataloader