在GPU上训练数据

模型搬到GPU上

数据搬到GPU上

损失函数计算搬到GPU上