fix type error

PaddlePaddle · Aug 9, 2023 · bfc390c · bfc390c
1 parent 6ae984f
commit bfc390c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/paddle/static/nn/common.py b/python/paddle/static/nn/common.py
@@ -24,7 +24,7 @@
     check_type,
     check_variable_and_dtype,
 )
-from paddle.fluid import core, layers, unique_name
+from paddle.fluid import core, unique_name
 from paddle.fluid.data_feeder import check_dtype
 from paddle.fluid.framework import (
     Program,
@@ -4210,7 +4210,7 @@ def update(self):
         Update Exponential Moving Average. Should only call this method in
         train program.
         """
-        global_step = layers.autoincreased_step_counter(
+        global_step = paddle.optimizer.lr.autoincreased_step_counter(
             counter_name=self._step_counter_name
         )
         param_master_emas = []

diff --git a/test/collective/fleet/parallel_dygraph_se_resnext.py b/test/collective/fleet/parallel_dygraph_se_resnext.py
@@ -66,7 +66,7 @@ def optimizer_setting(params, parameter_list=None):
             parameter_list=parameter_list,
         )
     else:
-        optimizer = fluid.optimizer.Momentum(
+        optimizer = paddle.optimizer.Momentum(
             learning_rate=paddle.optimizer.lr.cosine_decay(
                 learning_rate=lr, step_each_epoch=step, epochs=num_epochs
             ),