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