fix test_weight_decay_extend error (#28178)

PaddlePaddle · Oct 21, 2020 · 5d73bfd · 5d73bfd
1 parent 4873c20
commit 5d73bfd
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/python/paddle/fluid/contrib/tests/test_weight_decay_extend.py b/python/paddle/fluid/contrib/tests/test_weight_decay_extend.py
@@ -149,17 +149,20 @@ def check_weight_decay2(self, place, model):
 
             avg_cost = model(data, label, self.word_dict_len)
 
+            optimizer = fluid.optimizer.Adam(learning_rate=self.learning_rate)
+
+            params_grads = optimizer.backward(avg_cost)
+
             param_list = [(var, var * self.learning_rate)
                           for var in main_prog.block(0).all_parameters()]
 
-            optimizer = fluid.optimizer.Adam(learning_rate=self.learning_rate)
-
-            optimizer.minimize(avg_cost)
             for params in param_list:
                 updated_p = fluid.layers.elementwise_sub(
                     x=params[0], y=params[1])
                 fluid.layers.assign(input=updated_p, output=params[0])
 
+            optimizer.apply_optimize(avg_cost, startup_prog, params_grads)
+
             param_sum = self.run_program(place, [data, label])
         return param_sum