跳转到内容

流水线并行

把模型按层切成若干 stage 放到不同 GPU,微批次(micro-batch)依次流过。气泡(bubble)比例约为 (p1)/m(p-1)/m。本章比较 GPipe、1F1B、交错 1F1B 与零气泡调度。

  • 增大微批次数量 mm 可减小气泡,但增加激活内存
  • 1F1B 显著降低峰值激活内存
  • PP 的通信量最小,适合跨节点