模型的显存占用推理和训练阶段的显存需求不一样，精度不一样的占用也不同