return F.conv2d(input, weight, bias, self.stride,
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor
出现此类错误的原因主要是:
在训练模型时使用的gpu,而在预测时使用cup导致,所以需要将预测图片加上.cuda(),使用GPU进行预测
这样便可解决!