咖啡陪你

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