GPU现存消耗异常大

[BeyondLightYear]

使用demo_test.py 采用uhdm_checkpoint.pth进行推理测试。一张4096*3072尺寸的图像，现存消耗能达到32GB。搞不明白是什么原因，请帮忙指点。多谢多谢。