解决办法Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

NoSuchKey

猜你喜欢

转载自blog.csdn.net/qq_38703529/article/details/123703915