我在A100上跑该项目，结果发现在单张卡上显存占用达到了78G，这个占用太高了，能否优化？ 
