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之后就好了
【推荐】FlashTable:表单开发界的极速跑车,让你的开发效率一路狂飙
【推荐】Flutter适配HarmonyOS 5知识地图,实战解析+高频避坑指南
【推荐】博客园的心动:当一群程序员决定开源共建一个真诚相亲平台
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步