公开(公告)号：CN112541513B

公开(公告)日：2023-06-27

申请号：CN201910893080.0

申请日：2019-09-20

申请人： 百度在线网络技术(北京)有限公司

发明人： 谢德平 ,  钱玉磊 ,  贾荣来 ,  刘书聪 ,  胡明清 ,  丁瑞全 ,  刘林 ,  祁安龙 ,  王岳

IPC分类号： G06F18/214

摘要： 本申请公开了一种模型训练方法、装置、设备及存储介质，涉及人工智能技术领域。具体实现方案为：获取样本数据对应的各特征数据；从外存中查询各特征数据分别对应的权值参数，根据查询结果确定各特征数据的权值参数，并将各特征数据与权值参数对应存储至显存中；其中，权值参数表征对应特征数据的重要程度；通过GPU从显存中读取各特征数据和各特征数据的权值参数，利用读取的数据进行模型训练，并根据训练结果更新显存中的各权值参数；获取显存中更新后的权值参数，并利用更新后的权值参数更新外存中对应的权值参数。本申请实施例的技术方案减少了存储模型训练样本所需投入的内存资源，减少了资源开销，同时提升了所训练任务的稳定性。

公开(公告)号：CN112541513A

公开(公告)日：2021-03-23

申请号：CN201910893080.0

申请日：2019-09-20

申请人： 百度在线网络技术(北京)有限公司

发明人： 谢德平 ,  钱玉磊 ,  贾荣来 ,  刘书聪 ,  胡明清 ,  丁瑞全 ,  刘林 ,  祁安龙 ,  王岳

IPC分类号： G06K9/62

摘要： 本申请公开了一种模型训练方法、装置、设备及存储介质，涉及人工智能技术领域。具体实现方案为：获取样本数据对应的各特征数据；从外存中查询各特征数据分别对应的权值参数，根据查询结果确定各特征数据的权值参数，并将各特征数据与权值参数对应存储至显存中；其中，权值参数表征对应特征数据的重要程度；通过GPU从显存中读取各特征数据和各特征数据的权值参数，利用读取的数据进行模型训练，并根据训练结果更新显存中的各权值参数；获取显存中更新后的权值参数，并利用更新后的权值参数更新外存中对应的权值参数。本申请实施例的技术方案减少了存储模型训练样本所需投入的内存资源，减少了资源开销，同时提升了所训练任务的稳定性。