AttributeError: 'numpy.float64' object has no attribute 'cuda'

这种类型的错误都是由于没有转换为tensor张量导致的,只需要注意使用cuda()前通过torch.tensor()转为tensor就可以了,例如这样:resnet(Variable(torch.tensor(img).cuda()))
posted @ 2022-04-20 17:36  咖啡陪你  阅读(898)  评论(0编辑  收藏  举报