- 0
- 0
- 约3.08万字
- 约 24页
- 2023-05-18 发布于四川
- 举报
本公开涉及一种计算图的处理方法、装置、可读介质及电子设备,该计算图的处理方法通过获取待训练的机器学习模型对应的目标计算图,对该目标计算图进行节点融合处理,以得到包括多个子图的融合计算图;根据每种算子节点的多种预设切分方式,对该融合计算图中的算子节点进行有效切分,以得到该融合计算图对应的多种切分数据流图,该切分数据流图为带有有效切分方式的目标计算图;然后根据该切分数据流图搜索该目标计算图对应的并行配置策略,如此能够自动高效的生成由流水并行,数据并行以及模型并行三者混合并行的模型训练策略,能够有效提
(19)国家知识产权局
(12)发明专利申请
(10)申请公布号 CN 114841327 A
(43)申请公布日 2022.08.02
(21)申请号 202210594641.9
(22)申请日 2022.05.27
(71)申请人 北京字节跳动网络技术有限公司
地址
原创力文档

文档评论(0)