2021年9月
[源码解析] PyTorch 流水线并行实现 (4)--前向计算
摘要： 前几篇文章我们介绍了 PyTorch 流水线并行的基本知识，自动平衡机制和切分数据，本文我们结合论文内容来看看如何保证前向计算执行顺序。 阅读全文
posted @ 2021-09-28 21:24 罗西的思考 阅读(1645) 评论(0) 推荐(0)