将训练数据分割到不同设备,每个设备训练完整模型
将模型的不同层分配到不同设备,按顺序执行前向和反向传播
将模型的张量(如权重矩阵)按维度分割到不同设备
将模型的不同子模块(如注意力和前馈层)分配到不同设备
这道题你会答吗?花几分钟告诉大家答案吧!
扫描二维码,关注牛客网
下载牛客APP,随时随地刷题