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

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