[Bug]: ernie预训练报错

[ZTurboX]

软件环境

- paddlepaddle:2.5.1
- paddlepaddle-gpu: 2.5.1
- paddlenlp: develop

重复问题

• I have searched the existing issues

错误描述

用最新的paddlenlp develop增量预训练ernie-3.0-tiny-micro-v2-zh会报错TypeError: init() got an unexpected keyword argument 'enable_recompute'

稳定复现步骤 & 代码

训练脚本
python -u -m paddle.distributed.launch
--gpus "0,1"
--log_dir "output/ernie-3.0-tiny-micro-v2-zh/log"
run_pretrain.py
--model_type "ernie"
--model_name_or_path "./ernie-3.0-tiny-micro-v2-zh"
--tokenizer_name_or_path "./ernie-3.0-tiny-micro-v2-zh"
--input_dir "./train_data"
--output_dir "output/ernie-3.0-tiny-micro-v2-zh"
--split 949,50,1
--max_seq_len 512
--micro_batch_size 16
--use_amp true
--fp16_opt_level O2
--max_lr 0.0001
--min_lr 0.00001
--max_steps 1000000
--save_steps 50000
--checkpoint_steps 5000
--decay_steps 990000
--weight_decay 0.01
--warmup_rate 0.01
--grad_clip 1.0
--logging_freq 20
--num_workers 2
--eval_freq 1000
--device "gpu"
--share_folder false \

[w5688414]

升级一下paddle和paddlenlp试一试