From 84d54c2b54db6ce0aa03ec696cb484254ca4789f Mon Sep 17 00:00:00 2001
From: Bai Yifan <me@ethanbai.com>
Date: Fri, 22 Jan 2021 15:40:42 +0800
Subject: [PATCH] fix test_dygraph_quant_aware ut (#610)

---
 tests/dygraph/test_dygraph_quant_aware.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/dygraph/test_dygraph_quant_aware.py b/tests/dygraph/test_dygraph_quant_aware.py
index c354605f..524af89c 100644
--- a/tests/dygraph/test_dygraph_quant_aware.py
+++ b/tests/dygraph/test_dygraph_quant_aware.py
@@ -39,16 +39,16 @@ class ImperativeLenet(nn.Layer):
                 kernel_size=3,
                 stride=1,
                 padding=1),
-            paddle.nn.Pool2D(
-                pool_size=2, pool_type='max', pool_stride=2),
+            paddle.nn.AvgPool2D(
+                kernel_size=2, stride=2),
             paddle.nn.Conv2D(
                 in_channels=6,
                 out_channels=16,
                 kernel_size=5,
                 stride=1,
                 padding=0),
-            paddle.nn.Pool2D(
-                pool_size=2, pool_type='max', pool_stride=2))
+            paddle.nn.AvgPool2D(
+                kernel_size=2, stride=2))
 
         self.fc = paddle.nn.Sequential(
             paddle.nn.Linear(
-- 
GitLab