“test=develop”

9bb55120 · Joejiong · 739448de · 9bb55120
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

PaddleNLP/pretrain_language_models/XLNet/utils/init.py PaddleNLP/pretrain_language_models/XLNet/utils/init.py +1 -1

未找到文件。
--- a/PaddleNLP/pretrain_language_models/XLNet/utils/init.py
+++ b/PaddleNLP/pretrain_language_models/XLNet/utils/init.py
@@ -81,7 +81,7 @@ def init_pretraining_params(exe,
            return False

    fluid.io.load(
-        model_path=pretraining_params_path, program=main_program, executor=exe)
+        program=main_program, model_path=pretraining_params_path, executor=exe)

    if use_fp16:
        cast_fp32_to_fp16(exe, main_program)