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:我的程序的部分截
相关文章
相关标签/搜索