RuntimeError: Input type (torch.cuda.ByteTensor) and weight type (torch.cuda.FloatTensor) should be

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

如问题所示,你的输入的类型是字节型的tensor,而加载的权重的类型是float类型的tensor,需要将字节型的tensor转化为float型的tensor。eg:我的程序的部分截图,仅供参考。我当时是没有第18行的转换出现了该问题。

RuntimeError: Input type (torch.cuda.ByteTensor) and weight type (torch.cuda.FloatTensor) should be_第1张图片

你可能感兴趣的:(pytorch)