Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

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()