RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.cuda.ByteTensor for argument #3 ‘other’
在torch<1.0版本中torch.cuda.FloatTensor与torch.cuda.ByteTensor不能直接运算,将torch.cuda.ByteTensor转为float类型,加.float()即可
RuntimeError:Expected object of type torch.cuda.FloatTensor but found type torch.cuda.ByteTensor
最新推荐文章于 2022-04-10 17:17:44 发布