您的位置:首页 > 移动开发 > Objective-C

AttributeError: ‘numpy.ndarray‘ object has no attribute ‘cuda‘

2020-07-23 16:40 260 查看

这种类型的错误都是由于没有转换为tensor张量导致的,只需要注意使用cuda()前通过torch.tensor()转为tensor就可以了,例如这样:resnet(Variable(torch.tensor(img).cuda()))

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐