fix static save (#5091)

d0d39cc0 · liu zhengxi · GitHub · 89b9d0d6 · d0d39cc0
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

PaddleNLP/benchmark/transformer/static/train.py PaddleNLP/benchmark/transformer/static/train.py +1 -1

未找到文件。
--- a/PaddleNLP/benchmark/transformer/static/train.py
+++ b/PaddleNLP/benchmark/transformer/static/train.py
@@ -196,7 +196,7 @@ def do_train(args):
                if args.save_model:
                    model_path = os.path.join(
                        args.save_model, "step_" + str(step_idx), "transformer")
-                    paddle.io.save(train_program, model_path)
+                    paddle.static.save(train_program, model_path)

            batch_id += 1
            step_idx += 1