Update program.py (#13389)

refering to-https://www.paddlepaddle.org.cn/documentation/docs/en/api/paddle/incubate/ModelAverage_en.html it should be paddle.incunate.ModelAverage and paddle.incubate.optimizer.ModelAverage
2024-07-15 19:18:26 +05:30 · 2024-07-15 19:18:26 +05:30 · 5347be75f9
parent 603b3728ac
commit 5347be75f9
1 changed files with 1 additions and 1 deletions
--- a/tools/program.py
+++ b/tools/program.py
@ -442,7 +442,7 @@ def train(
                and dist.get_rank() == 0
            ):
                if model_average:
-                    Model_Average = paddle.incubate.optimizer.ModelAverage(
+                    Model_Average = paddle.incubate.ModelAverage(
                        0.15,
                        parameters=model.parameters(),
                        min_average_window=10000,