只需要将对应的torch.tensor()改为torch.FloatTensor()就可以了,例如torch.FloatTensor([img])
Input type (torch.cuda.ByteTensor) and weight type (torch.cuda.FloatTensor) should be the same
最新推荐文章于 2024-05-17 23:49:06 发布
只需要将对应的torch.tensor()改为torch.FloatTensor()就可以了,例如torch.FloatTensor([img])