PyTorch·torch.utils.checkpointv2.5.1用反向重计算减少训练激活占用+ 对比PyTorch 的非重入 activation checkpoint 在反向传播需要中间张量时重新执行对应前向片段,以额外计算换取更低的激活存储需求。激活重计算图附源码片段 · 内存管理