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:我的程序的部分截图
·
RuntimeError: Input type (torch.cuda.ByteTensor) and weight type (torch.cuda.FloatTensor) should be the same
如问题所示,你的输入的类型是字节型的tensor,而加载的权重的类型是float类型的tensor,需要将字节型的tensor转化为float型的tensor。eg:我的程序的部分截图,仅供参考。我当时是没有第18行的转换出现了该问题。
更多推荐
所有评论(0)