仅作为记录,大佬请跳过。

背景

程序运行到

        target = target.cuda()

时,出现错误AttributeError: 'tuple' object has no attribute 'cuda'

解决

tuple转成tensor

target是tuple类型,但.conda()需要是tensor类型

做法:

tuple——np.array——tensor(中间需要np.array中转;且np.array的元素需要是int或float(原本是str),使用.astype(int)转化)

即可。

展示:

        target=np.array(target).astype(int)
        target=torch.from_numpy(target)
        target = target.cuda()

参考

在这里插入图片描述
在这里插入图片描述

传送门


在这里插入图片描述
tuple->numpy传送门

在这里插入图片描述
numpy->tensor传送门

在这里插入图片描述

numpy数组中的元素全为int32整数型——传送门

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐