每次转发后如何释放临时消耗的GPU内存?我有这样的课程: class Stem(nn.Module): def __init__(self): super(Stem, self).__init__() ...2024-10-03 已阅读: n次