出现上述,是说torch.mm函数中出现问题,输入x需要转到gpu上,对将x.mm(self.w)变成torch.mm(x.to(device), self.w)即可。
pytorch Expected all tensors to be on the same device, but found at least two devices, cpu
最新推荐文章于 2024-08-30 20:31:35 发布
出现上述,是说torch.mm函数中出现问题,输入x需要转到gpu上,对将x.mm(self.w)变成torch.mm(x.to(device), self.w)即可。