希望增加uie模型微调训练时，爆显存的提示

[dedsecezio]

问题描述

没有显存报错信息：
我在按照文本信息抽取readme进行配置小样本微调时，运行#2.3 finetune.py后没有保存模型，仅仅在checkpoint目录下有log。
debug和检查环境配置花了很多时间发现电脑8G显存不够，运行时爆显存需要降低batch_size，但是没有任何报错或是提示，只是直接结束运行。

[w5688414]

显存和内存不够可以使用uie-small模型试一下，另外，uie已经停止更新了，欢迎开发者进行贡献。