pytorch 在用gpu训练模型训练 损失函数不需要迁移到gpu