激活重计算(Gradient Checkpointing)的原理?

一句话速记:前向只保存部分层的输入,反向时重新计算中间激活,用约 30% 额外计算换大量显存。

显存

同章节的其他八股