南方电网申请模型训练方法装置和计算机设备专利,提高模型训练效率

摘要:国家知识产权局信息显示,南方电网人工智能科技有限公司申请一项名为“模型训练方法装置和计算机设备”的专利,公开号CN 119067173 A,申请日期为2024年8月。

金融界2024年12月5日消息,国家知识产权局信息显示,南方电网人工智能科技有限公司申请一项名为“模型训练方法装置和计算机设备”的专利,公开号CN 119067173 A,申请日期为2024年8月。

专利摘要显示,本申请涉及一种模型训练方法、装置和计算机设备,从图形处理器GPU中,获取训练模型的本轮次梯度数据;获取的本轮次梯度数据的数据格式为整数型;并根据本轮次梯度数据和训练模型的本轮次模型参数,确定训练模型的下一轮次模型参数;进而将下一轮次模型参数发送至GPU,以使GPU基于下一轮次模型参数进行前向传播和反向传播,得到训练模型的下一轮次梯度数据。上述方案,GPU与CPU之间进行各轮次梯度数据传播时,各轮次梯度数据为整数型,缩减了数据传输量,提高了数据传输效率,进而提高了模型训练的效率。

来源:金融界

相关推荐