[发明专利]训练任务的管理方法、装置、电子设备及存储介质

说明书

本公开提供了一种训练任务的管理方法、装置、电子设备及存储介质，其中，该方法包括：响应于任务创建指令，展示任务创建界面，任务创建界面包含用于创建目标训练任务的多个任务配置选项；响应于针对多个任务配置选项的选填操作，获取目标训练任务的任务属性信息以及获取资源属性信息；资源属性信息包括执行目标训练任务所需要的目标资源池，资源池用于指示执行目标训练任务所对应的目标数据设备集群中的一组目标数据设备；在虚拟资源池中创建第一任务对象以存储任务属性信息以及资源属性信息；第一任务对象用于指示目标数据设备集群中的目标数据设备执行目标训练任务的训练操作。本公开实施例，可以提升训练任务在多集群下的部署效率和准确度。

技术领域

本公开涉及人工智能技术领域，具体而言，涉及一种训练任务的管理方法、训练任务的管理装置、电子设备及计算机可读存储介质。

背景技术

随着人工智能的飞速发展，基于数据进行训练得到的深度学习模型(算法模型)在多个领域得到应用，例如常见的机器翻译、语义挖掘、图像识别、人脸识别、语音识别等领域。

目前，在进行算法模型的训练时，为了提升训练任务的训练效率，通常采用分布式的方式来进行任务训练，也即，通过多台主机的资源来共同辅助训练一个训练任务。然而，面对分布式训练，如何实现任务的有效部署，是业界一直追求的目标。

发明内容

本公开实施例至少提供一种训练任务的管理方法、训练任务的管理装置、电子设备及计算机可读存储介质。

本公开实施例提供了一种训练任务的管理方法，包括：

响应于任务创建指令，展示任务创建界面，所述任务创建界面包含用于创建目标训练任务的多个任务配置选项；

响应于针对所述多个任务配置选项的选填操作，获取所述目标训练任务的任务属性信息以及获取资源属性信息；其中，所述资源属性信息包括执行所述目标训练任务所需要的目标资源池，所述资源池用于指示执行所述目标训练任务所对应的目标数据设备集群中的一组目标数据设备；

响应于训练任务启动指令，确定与所述目标资源池对应的虚拟资源池，并在所述虚拟资源池中创建第一任务对象以存储所述任务属性信息以及所述资源属性信息；所述第一任务对象用于指示所述目标数据设备集群中的目标数据设备执行所述目标训练任务的训练操作。

本公开实施例中，能够为用户展示任务创建界面，且在用户选填任务创建界面中多个任务配置选项后，即可以获取到目标训练任务的任务属性信息以及获取资源属性信息，进而可以确定执行该目标训练任务所对应的目标数据设备集群中的一组目标数据设备，再通过创建任务对象的方式指示所述目标数据设备集群中的目标数据设备执行所述目标训练任务的训练操作即可，如此，不仅实现了目标训练任务在目标数据设备上的部署，还在方便用户创建训练任务的同时提升了任务部署的效率和准确率。

在一种可能的实施方式中，所述任务创建界面包含资源池选择配置项，所述获取资源属性信息，包括：

获取目标用户的身份信息，并基于目标用户的身份信息确定与所述目标用户关联的至少一个资源池；

响应于所述目标用户针对所述资源池选择配置项的选择操作，从所述至少一个资源池中确定所述目标资源池。

本公开实施例中，根据用户的身份信息确定与目标用户关联的至少一个资源池，并基于用户的选择从所述至少一个资源池中确定所述目标资源池，如此，可以保证用户提交的目标训练任务会在该用户关联的资源池上执行，有利于实现资源池的有效利用。

在一种可能的实施方式中，所述任务创建界面包含容器资源类型配置项，所述资源属性信息还包括资源类型，所述获取所述资源属性信息包括：

基于目标用户针对所述容器资源类型配置项填写确定操作，从所述目标资源池中确定执行所述目标训练任务所需要的资源类型；其中，所述资源类型包括中央处理器资源、图形处理器资源以及内存资源中的至少一种。