【PIR API adaptor No.187】 Migrate paddle.regularizer.L2Decay into pir (#59313)

xiaoyewww · web-flow · commit bb190f1c38d8 · 2023-11-28T11:23:59.000+08:00
diff --git a/test/legacy_test/test_regularizer_api.py b/test/legacy_test/test_regularizer_api.py
@@ -22,6 +22,7 @@
 import paddle
 from paddle import base
 from paddle.base import core
+from paddle.pir_utils import test_with_pir_api
 
 
 def bow_net(
@@ -102,8 +103,8 @@ def run_program(self, place, feed_list):
     def check_l2decay_regularizer(self, place, model):
         paddle.seed(1)
         paddle.framework.random._manual_program_seed(1)
-        main_prog = base.framework.Program()
-        startup_prog = base.framework.Program()
+        main_prog = paddle.static.Program()
+        startup_prog = paddle.static.Program()
         with self.scope_prog_guard(
             main_prog=main_prog, startup_prog=startup_prog
         ):
@@ -175,16 +176,20 @@ def test_l2(self):
                     rtol=5e-5,
                 )
 
+    @test_with_pir_api
     def test_repeated_regularization(self):
         paddle.enable_static()
         l1 = paddle.regularizer.L1Decay(0.1)
         l2 = paddle.regularizer.L2Decay(0.01)
         fc_param_attr = paddle.ParamAttr(
             regularizer=paddle.regularizer.L1Decay()
         )
-        with base.program_guard(base.Program(), base.Program()):
+        with base.program_guard(
+            paddle.static.Program(), paddle.static.Program()
+        ):
             x = paddle.uniform([2, 2, 3])
-            out = paddle.static.nn.fc(x, 5, weight_attr=fc_param_attr)
+            linear = paddle.nn.Linear(3, 5, weight_attr=fc_param_attr)
+            out = linear(x)
             loss = paddle.sum(out)
             sgd = paddle.optimizer.SGD(learning_rate=0.1, weight_decay=l2)
             sgd.minimize(loss)
