Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
翻译为:输入类型(torch.FLOTTENSOR)和重量类型(torch.cuda.FLOTTENSOR)应相同
这一般原因是模型和输入参数只有一个设置了显卡来加速,忘记另外一个也设置显卡加速。
修改:
#模型
tu().cuda()
#输入图片
if torch.cuda.is_available():
imgs = imgs.cuda()
targets = targets.cuda()