预训练的时候在加载模型的过程中卡住

[amanyara]

Describe the bug
A clear and concise description of what the bug is.
在我预训练模型的时候，模型加载的一晚上还在加载。显存没有跑满。

To Reproduce
Steps to reproduce the behavior:

1. cd /home/pengc/ernie/ernie-develop/demo
2. python pretrain.py --data_dir "../data/*.gz" --from_pretrained ../ernie-gram-zh --save_dir ./outabtrain_1012

Expected behavior
A clear and concise description of what you expected to happen.
期待模型开始训练，日志输出内容

Screenshots
If applicable, add screenshots to help explain your problem.

Additional context
Add any other context about the problem here.
运行环境：Ubuntu18.04
显卡：A800*2
Python：3.8.18
cuDNN：8.4
PaddlePaddle-GPU： 2.5.1.post120
运行指令：如图