想把代码放到GPU上跑,将数据集转移到CUDA上出现错误‘list’object has no attribute ‘cuda’
无脑百度
原因是list列表没有CUDA属性
无脑通过numpy转化为tensor 后
报错ValueError: only one element tensors can be converted to Python scalars
仔细看了看,自己数据是这样的 [tensor,tensor,tensor]
因此
targets =[ target.cuda() for target in targets]
解决
很简单,希望自己以后遇到问题先动脑子