报错的意思是类型不匹配,然后通过debug发现是float32,一个是float64,虽然都是浮点数,但就是不一样!
解决办法:在这两个矩阵后面都加上.float()。
这样就都是float32了!再次运行就成功了!
x = torch.from_numpy(digits.data).float()
y = torch.from_numpy(digits.target).float().view(360, -1)
报错的意思是类型不匹配,然后通过debug发现是float32,一个是float64,虽然都是浮点数,但就是不一样!
解决办法:在这两个矩阵后面都加上.float()。
这样就都是float32了!再次运行就成功了!
x = torch.from_numpy(digits.data).float()
y = torch.from_numpy(digits.target).float().view(360, -1)