GUP训练 配置pytorch的gup版本主要是在网络模型、输入和标记的数据、损失函数 方式一 直接.cuda()调用,在原有的模型训练代码中的网络模型、输入和标记的数据、损失函数部分直接调用即可 方式二 事先定义好设备device,然后直接.to(device)调用,在原有的模型训练代码中的网络模型、输入和标记的数据、损失函数部分直接调用即可 训练代码部分可参考上一篇:Pytorch学习笔记(模型训练)