[发明专利]模型训练方法、装置、电子设备、存储介质和开发系统

权利要求书

1.一种模型训练方法，包括：

根据确定接收到训练准备指令，调用训练准备组件来至少设置用于训练所述模型的损失函数和优化函数；以及

根据确定接收到训练指令，调用训练组件来设置第一数据读取组件，所述第一数据读取组件被配置为载入用于训练所述模型的训练数据集；以及

通过所述训练组件利用所述损失函数和所述优化函数，基于来自所述第一数据读取组件的所述训练数据集，训练所述模型。

2.根据权利要求1所述的方法，其中所述方法还包括：

调用所述训练准备组件来设置度量函数；

根据确定接收到评估指令，调用评估组件以设置第二数据读取组件，所述第二数据读取组件被配置为载入不同于所述训练数据集的评估数据集；以及

通过所述评估组件利用经训练的所述模型以及所述损失函数和所述度量函数，基于来自所述第二数据读取组件的所述评估数据集，评估所述模型。

3.根据权利要求1所述的方法，进一步包括：

根据确定接收到预测指令，调用预测组件来设置第三数据读取组件，所述第三数据读取组件被配置为载入不同于所述训练数据集的预测数据集；以及

通过所述预测组件利用经训练的所述模型，确定来自所述第三数据读取组件的所述预测数据集的预测结果。

4.根据权利要求1所述的方法，还包括：

获取并行处理所述训练数据集的计算设备的数目；

基于所述数目对所述训练数据集进行分组，以将所述训练数据集中的训练数据布置在所述第一数据读取组件中；

确定所述第一数据读取组件中缺少训练数据的位置；以及

以预定数据填充所述位置。

5.根据权利要求2所述的方法，还包括：

获取并行处理所述评估数据集的计算设备的数目；

基于所述数目对所述评估数据集进行分组，以将所述评估数据集中的评估数据布置在所述第二数据读取组件中；

确定所述第二数据读取组件中缺少评估数据的位置；以及

以预定数据填充所述位置，以形成经填充的所述评估数据集。

6.根据权利要求5所述的方法，还包括：

利用经训练的所述模型以及经设置的所述损失函数和所述度量函数，基于经填充的所述评估数据集，确定所述模型的评估结果；

最小化所述评估结果中与缺少评估数据的所述位置相对应的无效评估结果的权重，以去除所述无效评估结果。

7.根据权利要求1所述的方法，其中所述训练组件具有分别针对所述损失函数、所述优化函数和所述第一数据读取组件的调用路径。

8.根据权利要求2所述的方法，其中所述评估组件具有分别针对所述损失函数、所述度量函数和所述第二数据读取组件的调用路径。

9.一种模型训练装置，包括：

训练准备组件调用模块，被配置为根据确定接收到训练准备指令，调用训练准备组件来至少设置用于训练所述模型的损失函数和优化函数；以及

训练组件调用模块，被配置为根据确定接收到训练指令，调用训练组件来设置第一数据读取组件，所述第一数据读取组件被配置为载入用于训练所述模型的训练数据集；以及

训练执行模块，被配置为通过所述训练组件利用所述损失函数和所述优化函数，基于来自所述第一数据读取组件的所述训练数据集，训练所述模型。

10.根据权利要求9所述的装置，其中所述训练准备组件调用模块进一步被配置为调用所述训练准备组件来设置度量函数，所述装置进一步包括：

评估组件调用模块，被配置为根据确定接收到评估指令，调用评估组件以设置第二数据读取组件，所述第二数据读取组件被配置为载入不同于所述训练数据集的评估数据集；以及

评估执行模块，被配置为通过所述评估组件利用经训练的所述模型以及所述损失函数和所述度量函数，基于来自所述第二数据读取组件的所述评估数据集，评估所述模型。