torch.from_numpy()报错:TypeError: expected np.ndarray (got numpy.ndarray)
看到有人说把torch.from_numpy()改成torch.Tensor(),我试了下确实可以,但是仅限于你只有这一个地方报错,如果用到torchvision.transforms之类的库,只要里面有从numpy转torch的操作就会报错
后来发现是因为numpy版本太高,我的是2.0.0,改成1.16.4之后就好了
看到有人说把torch.from_numpy()改成torch.Tensor(),我试了下确实可以,但是仅限于你只有这一个地方报错,如果用到torchvision.transforms之类的库,只要里面有从numpy转torch的操作就会报错
后来发现是因为numpy版本太高,我的是2.0.0,改成1.16.4之后就好了