From fa6cf5c7fab5ea94a5db69861535d0ff27514061 Mon Sep 17 00:00:00 2001
From: pengjuncai <13006307475@163.com>
Date: Thu, 18 Mar 2021 05:05:22 +0000
Subject: [PATCH] up, test=develop

---
 .../fluid/contrib/slim/quantization/imperative/utils.py     | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/paddle/fluid/contrib/slim/quantization/imperative/utils.py b/python/paddle/fluid/contrib/slim/quantization/imperative/utils.py
index 000152f272785..1ff4a408e051f 100644
--- a/python/paddle/fluid/contrib/slim/quantization/imperative/utils.py
+++ b/python/paddle/fluid/contrib/slim/quantization/imperative/utils.py
@@ -54,6 +54,6 @@
 
 out_scale_layers_list = (
     paddle.nn.Conv2D, paddle.nn.Linear, paddle.nn.MaxPool2D,
-    paddle.nn.BatchNorm, paddle.nn.SyncBatchNorm, paddle.nn.LeakyReLU,
-    paddle.nn.PReLU, paddle.nn.ReLU, paddle.nn.ReLU6, paddle.nn.Sigmoid,
-    paddle.nn.Softmax, paddle.nn.Tanh, paddle.nn.Swish)
+    paddle.nn.BatchNorm, paddle.nn.BatchNorm2D, paddle.nn.SyncBatchNorm,
+    paddle.nn.LeakyReLU, paddle.nn.PReLU, paddle.nn.ReLU, paddle.nn.ReLU6,
+    paddle.nn.Sigmoid, paddle.nn.Softmax, paddle.nn.Tanh, paddle.nn.Swish)