面向模型并行训练的模型拆分策略自动生成方法

王丽, 郭振华, 曹芳, 高开, 赵雅倩, 赵坤

An automatic model splitting strategy generation method for model parallel training

WANG Li, GUO Zhen-hua, CAO Fang, GAO Kai, ZHAO Ya-qian, ZHAO Kun

计算机工程与科学 . 2020, (09): 1529 -1537 .