预训练模型的获取方法、装置以及计算机设备

基本信息

申请号 202011204435X 申请日 -
公开(公告)号 CN112288098A 公开(公告)日 2021-01-29
申请公布号 CN112288098A 申请公布日 2021-01-29
分类号 G06N20/00(2019.01)I 分类 计算;推算;计数;
发明人 钱勇;付冰冰;方杰 申请(专利权)人 平安数字信息科技(深圳)有限公司
代理机构 深圳市明日今典知识产权代理事务所(普通合伙) 代理人 王杰辉;宋庆洪
地址 518000广东省深圳市前海深港合作区南山街道临海大道59号海运中心主塔楼1411号-14158
法律状态 -

摘要

摘要 本发明提供了一种预训练模型的获取方法、装置以及计算机设备,其中方法包括:获取来自第一终端的终端初始模型;根据所述终端初始模型的类别,从所述终端初始模型所属的目录类别下,筛选出多个暂时初始模型;输入相同的训练数据至每个所述暂时初始模型中进行训练;根据各训练好的预训练模型的所述损失值挑选对应的所述预训练模型发送给所述第一终端。本发明的有益效果:实现了自动根据开发者提供的终端初始模型搜寻相似的初始模型一起训练,训练完成后自动计算损失值,根据损失值为开发者挑选出合适的预训练模型,提高了开发者的开发效率,以及提高模型训练准确度。