remove with_data_parallel in static graph training (#5438)

* remove with_data_parallel in static graph training * fix codestyle
PaddlePaddle · Apr 3, 2023 · 5931590 · 5931590
1 parent 2560f2d
commit 5931590
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/examples/machine_translation/transformer/static/train.py b/examples/machine_translation/transformer/static/train.py
@@ -228,9 +228,7 @@ def do_train(args):
         build_strategy = paddle.static.BuildStrategy()
         exec_strategy = paddle.static.ExecutionStrategy()
 
-        compiled_train_program = paddle.static.CompiledProgram(train_program).with_data_parallel(
-            loss_name=avg_cost.name, build_strategy=build_strategy, exec_strategy=exec_strategy
-        )
+        compiled_train_program = paddle.static.CompiledProgram(train_program, build_strategy=build_strategy)
     exe.run(startup_program)
 
     if args.use_amp: