摘要: 看到有人说把torch.from_numpy()改成torch.Tensor(),我试了下确实可以,但是仅限于你只有这一个地方报错,如果用到torchvision.transforms之类的库,只要里面有从numpy转torch的操作就会报错 后来发现是因为numpy版本太高,我的是2.0.0,改成 阅读全文
posted @ 2024-07-05 10:11 钢之炼丹术师 阅读(1882) 评论(0) 推荐(0)