Runtime: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or
module = torch.load("vincent_0.pth", map_location=torch.device("cpu"))
原创
2022-04-18 17:24:36 ·
1059 阅读 ·
0 评论