diff --git a/PaddleNLP/pretrain_language_models/XLNet/utils/init.py b/PaddleNLP/pretrain_language_models/XLNet/utils/init.py index 0ad4f31b161bd6cca4f684ff1ae101a12a80f7a9..fad4d687085d49e843e878391976371f915d6320 100644 --- 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)