Input type (torch.cuda.ByteTensor) and weight type (torch.cuda.FloatTensor) should be the same 只需要将对应的torch.tensor()改为torch.FloatTensor()就可以了，例如torch.FloatTensor([img]) 代码 2023-10-01 45 点赞 0 评论 68 浏览
