利用强化学习的设备布局优化.pdfVIP

  • 1
  • 0
  • 约2.01万字
  • 约 17页
  • 2024-03-06 发布于四川
  • 举报
本申请涉及利用强化学习的设备布局优化。描述了一种用于确定跨多个硬件设备的机器学习模型操作的布局的方法。该方法包括接收指定将被布局用于在多个硬件设备上进行分布式处理的机器学习模型的数据;从数据生成操作嵌入序列,序列中的每个操作嵌入表征执行机器学习模型的处理所需的相应操作;根据布局递归神经网络的多个网络参数的第一值,使用布局递归神经网络处理操作嵌入序列,以生成定义由序列中的操作嵌入表征的跨多个设备的操作的布局的网络输出;以及通过根据由网络输出定义的布局在多个设备上布局操作,来调度机器学习模型以供多个

(19)国家知识产权局

(12)发明专利申请

(10)申请公布号CN117648971A

(43)申请公布日2024.03.05

(21)申请号202311517807.8(51)Int.Cl.

文档评论(0)

1亿VIP精品文档

相关文档