摘要
本申请涉及深度学习技术领域,具体公开了一种基于昇腾平台的模型训练框架适配方法、装置、计算机设备及存储介质。获取当前模型训练框架与昇腾平台中Mode l Li nk训练框架的待适配内容,并对当前模型训练框架进行调整及迁移,获得初始模型训练框架;基于框架集成指令,将昇腾平台中的Mi ndSpeed加速框架集成到初始模型框架中,获得待验证模型训练框架;基于预训练数据集和实验数据集,进行模型预训练和正式训练,获得模型训练指标;在模型训练指标符合指标要求时,获得基于昇腾平台的目标模型训练框架。本申请通过将当前模型训练框架迁移适配到昇腾平台上,无需借助原有国外平台,降低了模型训练成本;其次,昇腾平台支持大规模扩展,提高了模型训练规模。