买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:深圳市欢太科技有限公司
摘要:本申请公开一种搜索方法、搜索装置、计算机设备和非易失性计算机可读存储介质。搜索方法包括根据待训练模型运行时的真实参数量和激活值,确定待训练模型的总显存占用量;根据各个并行组合参数和总显存占用量,确定各个并行组合参数对应的单卡显存占用量,多个并行组合参数由预设的多个并行参数组合而成;在多个并行组合参数中,筛选出单卡显存占用量小于当前计算机设备的单卡存储上限的多个目标并行组合参数。如此,在多个并行组合参数中筛选出符合要求的并行组合参数,能够节省后续进行最优批量大小搜索,以获取最优训练配置所需的时间,从而提高待训练模型训练的效率。
主权项:1.一种搜索方法,其特征在于,包括:根据待训练模型运行时的真实参数量和激活值,确定待训练模型的总显存占用量;根据各个并行组合参数和所述总显存占用量,确定各个所述并行组合参数对应的单卡显存占用量,多个所述并行组合参数通过预设的多个并行参数中的取值进行组合得到;在多个所述并行组合参数中,筛选出所述单卡显存占用量小于当前计算机设备的单卡存储上限的多个目标并行组合参数。
全文数据:
权利要求:
百度查询: 深圳市欢太科技有限公司 搜索方法、搜索装置、计算机设备及存储介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。