会员
周边
捐助
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
咖啡陪你
博客园
首页
新随笔
联系
订阅
管理
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
)
编辑
收藏
举报
刷新页面
返回顶部
公告