基于昇腾平台的模型训练框架适配方法及装置

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
基于昇腾平台的模型训练框架适配方法及装置
申请号:CN202510517064
申请日期:2025-04-23
公开号:CN120611181A
公开日期:2025-09-09
类型:发明专利
摘要
本申请涉及深度学习技术领域,具体公开了一种基于昇腾平台的模型训练框架适配方法、装置、计算机设备及存储介质。获取当前模型训练框架与昇腾平台中Mode l Li nk训练框架的待适配内容,并对当前模型训练框架进行调整及迁移,获得初始模型训练框架;基于框架集成指令,将昇腾平台中的Mi ndSpeed加速框架集成到初始模型框架中,获得待验证模型训练框架;基于预训练数据集和实验数据集,进行模型预训练和正式训练,获得模型训练指标;在模型训练指标符合指标要求时,获得基于昇腾平台的目标模型训练框架。本申请通过将当前模型训练框架迁移适配到昇腾平台上,无需借助原有国外平台,降低了模型训练成本;其次,昇腾平台支持大规模扩展,提高了模型训练规模。
技术关键词
框架 模型预训练 指标 策略 计算机设备 性能分析工具 数据分布 平台专用 深度学习技术 可读存储介质 预训练模型 存储计算机程序 场景 监控工具 适配装置 处理器 数据平台 日志