问题描述:利用pytorch进行模型训练时,在加载数据后出现如下报错
inputs, gts = inputs.cuda(), gts.cuda()
AttributeError: ‘numpy.ndarray‘ object has no attribute ‘cuda‘
问题解决:
这种类型的错误都是由于没有转换为tensor张量导致的,只需要注意使用cuda()前通过torch.tensor()转为tensor就可以了,例如这样:
inputs, gts = torch.tensor(inputs).cuda(), torch.tensor(gts).cuda()
参考博客:https://blog.csdn.net/qq_34124009/article/details/107541222