Samar-blog

摘要： 1.代码实战： （1）调用.to()方法即可。 .to(device) device = torch.device("cpu") torch.device("cuda") 点击查看代码 #12.定义训练的设备 # device = torch.device("cpu") device = torch 阅读全文

摘要： 1.方法1：调用.cuda()方法 （1）找到网络模型、数据（输入、标注）和损失函数，然后调用其.cuda()方法： ①网络模型如dyl模型：dyl.cuda()，②数据（输入、标注）即：imgs = imgs.cuda()，targets = targets.cuda()，③损失函数即loss_f 阅读全文