【技术实现步骤摘要】
模型训练方法、装置及计算机存储介质
[0001]本申请涉及大数据领域,尤其涉及一种模型训练方法、装置及计算机存储介质。
技术介绍
[0002]在网页中推广广告时,往往需要预测用户对各个广告的点击率,并将预测的点击率靠前的广告显示在网页中显眼的位置,以达到广告推广的效果。其中,用户对各个广告的点击率通常通过深度模型来预测。开发者一般会选择TensorFlow作为模型的训练框架。但TensorFlow的训练速度慢、执行效率低。
技术实现思路
[0003]本申请实施例提供了一种模型训练方法、装置及计算机存储介质,可以提升模型的训练速度,提高执行效率。
[0004]第一方面,本申请实施例提供了一种模型训练方法,包括:
[0005]获取至少一个待更新的模型参数;所述至少一个待更新的模型参数为模型参数的一部分;
[0006]将所述至少一个待更新的模型参数输入到模型中,获取到梯度值以更新所述至少一个待更新的模型参数。
[0007]本申请实施例中通过仅将当前训练所需更新的模型参数输入到模型中进 ...
【技术保护点】
【技术特征摘要】
1.一种模型训练方法,其特征在于,包括:获取至少一个待更新的模型参数;所述至少一个待更新的模型参数为模型参数的一部分;将所述至少一个待更新的模型参数输入到模型中,获取到梯度值以更新所述至少一个待更新的模型参数。2.如权利要求1所述的方法,其特征在于,所述至少一个待更新的模型参数分别存储在多个节点中。3.如权利要求2所述的方法,其特征在于,所述至少一个待更新的模型参数按照哈希值均匀分布在所述多个节点中。4.如权利要求2所述的方法,其特征在于,所述获取至少一个待更新的模型参数之前,所述方法还包括:查找所述至少一个待更新的模型参数各自对应的存储节点;所述获取至少一个待更新的模型参数包括:分别从所述至少一个待更新的模型参数各自对应的存储节点获取所述至少一个待更新的模型参数。5.如权利要求4所述的方法,其特征在于,所述查找至少一个待更新的模型参数的存储节点之前,所述方法还包括:确定所述至少一个待更新的模型参数的哈希值;所述查找至少一个待更新的模型参数的存储节点,包括:根据所述至少一个待更新的模型参数的哈希值,确定所述至少一个待更新的模型参数各自对应的存储节点。6.如权利要求2所述的方法...
【专利技术属性】
技术研发人员:张彦升,姚磊,
申请(专利权)人:北京鸿享技术服务有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。