摘要
本公开涉及一种模型训练方法及装置、电子设备和存储介质,所述方法包括:根据第一模型训练框架、第一GPU对应的第一软件栈、以及预设模型训练参数,构建Docker镜像,第一模型训练框架为调用第一GPU进行模型训练的模型训练框架,第一模型训练框架是对第二模型训练框架进行参数转换后得到的,第二模型训练框架为调用第二GPU进行模型训练的模型训练框架;将Docker镜像发送至多个训练节点,以及控制每个训练节点基于Docker镜像创建Docker容器;针对任意一个训练节点,控制该训练节点在Docker容器内调用该训练节点上部署的第一GPU进行模型预训练。本公开实施例可以有效实现模型训练的快速部署、快速训练。
技术关键词
框架
软件
模型预训练
模型训练方法
应用程序编程接口
节点
镜像
计算机程序指令
容器
模型训练装置
参数
电子设备
标签
处理器
模块
可读存储介质
存储器
精度
系统为您推荐了相关专利信息
神经网络模型
电子元器件
布局
CAE仿真技术
文件夹
风险预测方法
卷积神经网络模型
时间序列预测模型
渔网
核心
机械臂控制方法
汉字书法
汉字字库
汉字专用
笔画