技术总结
本公开是关于一种确定硬件运算平台分配方式的方法和装置,属于机器学习模型应用技术领域。所述方法包括:确定待运行的机器学习模型中包括的多个计算节点;获取多个计算节点对应的多组样本平台配置信息;对于每组样本平台配置信息,基于样本平台配置信息控制将多个计算节点分配到各硬件运算平台,控制各硬件运算平台执行分配到的各计算节点,统计各硬件运算平台执行多个计算节点的总执行时长,作为样本平台配置信息对应的总执行时长;基于多组样本平台配置信息和分别对应的总执行时长,确定优化平台配置信息。采用本公开,可以保证异构硬件设备能够以较短的整体执行时长快速运行机器学习模型对应的运行代码。
技术研发人员:韩新承
受保护的技术使用者:杭州海康威视数字技术股份有限公司
技术研发日:2019.12.16
技术公布日:2021.06.18