GPU occupy

[ZoeSnow]

这个代码中有哪部分是能够减少显存占用吗，为什么我自己的模型训练时占满GPU，替换restormer的模型部分放到这个代码里训练时只占用几个G。

[Mytttttttt]

这是来自QQ邮箱的假期自动回复邮件。你好，我最近正在休假中，无法亲自回复你的邮件。我将在假期结束后，尽快给你回复。