[发明专利]一种模型训练的保护方法、装置、电子设备及存储介质

权利要求书

1.一种模型训练的保护方法，其特征在于，包括以下步骤：

以预设周期获取模型信息，保存至共享内存；

在模型训练的过程中监测终止信号；

当监测到终止信号时，获取共享内存中的最新模型信息，所述共享内存用于存储训练模型时的模型信息；判断终止类型，所述终止类型包括正常终止、用户终止、程序异常终止以及系统异常终止；当终止类型为用户终止、程序异常终止时，获取模型信息，保存至共享内存；

将所述最新模型信息保存为模型恢复文件；

根据模型恢复文件进行模型的恢复；

所述模型信息包括：

在训练阶段保存超网络参数、采样参数以及训练的期数；

在训练结束时保存超网络参数、备选网络集数组列表以及精度。

2.根据权利要求1所述的模型训练的保护方法，其特征在于，所述在模型训练的过程中监测终止信号的步骤之前，还包括：

根据配置文件设定保存模型的位置及文件名。

3.一种模型训练的保护装置，其特征在于，包括：

信号获取模块，用于以预设周期获取模型信息，保存至共享内存；在模型训练的过程中监测终止信号；

模型信息获取模块，用于当监测到终止信号时，获取共享内存中的最新模型信息，所述共享内存用于存储训练模型时的模型信息；判断终止类型，所述终止类型包括正常终止、用户终止、程序异常终止以及系统异常终止；当终止类型为用户终止、程序异常终止时，获取模型信息，保存至共享内存；

模型信息保存模块，用于将所述最新模型信息保存为模型恢复文件；

模型恢复模块，用于根据模型恢复文件进行模型的恢复；

所述模型信息包括：

在训练阶段保存超网络参数、采样参数以及训练的期数；

在训练结束时保存超网络参数、备选网络集数组列表以及精度。

4.一种电子设备，包括存储器、处理器，所述存储器中存储有可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现上述权利要求1至2任一项所述的方法的步骤。

5.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有机器可运行指令，所述计算机可运行指令在被处理器调用和运行时，所述计算机可运行指令促使所述处理器运行所述权利要求1至2任一项所述的方法。