这种类型的错误都是由于没有转换为tensor张量导致的,只需要注意使用cuda()前通过torch.tensor()转为tensor就可以了,例如这样:resnet(Variable(torch.tensor(img).cuda()))
AttributeError: ‘numpy.ndarray‘ object has no attribute ‘cuda‘
最新推荐文章于 2022-05-15 14:30:33 发布
这种类型的错误都是由于没有转换为tensor张量导致的,只需要注意使用cuda()前通过torch.tensor()转为tensor就可以了,例如这样:resnet(Variable(torch.tensor(img).cuda()))