一种任务分配方法、装置及电子设备和存储介质

专利主权项内容

1.一种任务分配方法，其特征在于，包括：获取分布式训练的物理节点集合、数据源集合、神经网络模型、事件树集合；其中，所述事件树集合的事件树基于所述神经网络模型的层拓扑结构和所述神经网络模型的每个层包含的事件创建，所述事件树用于描述数据源与事件或不同事件之间的关联关系；根据所述神经网络模型的每个层传出数据与传入数据的比率和处理单位流量所需的计算资源确定多个所述事件树的总能耗，选择总能耗最小的事件树作为目标事件树；以最小化训练能耗为目标确定所述目标事件树的目标事件或数据源分配策略；其中，所述目标事件或数据源分配策略用于描述处理所述目标事件树中事件的物理节点或数据源分配的物理节点，所述训练能耗包括所述物理节点处理事件的能耗、不同事件之间的数据传输能耗；基于所述目标事件或数据源分配策略将所述数据源和所述目标事件树中的事件部署于对应的物理节点。