Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

人工智能领域新星创作者

2022-10-26 09:42:29

输入的是gpu的torch，但是权重weight是cpu的，所以部匹配报错

重新到处模型使用gpu导出：

python export.py --weights runs/train/exp/weights/best.pt --include torchscript --device 0

...全文

284 回复打赏收藏转发到动态举报

写回复

回复

切换为时间正序

请发表友善的回复…

发表回复

YOLOX是旷视科技新近推出的高性能实时目标检测网络，性能超越了YOLOv3/YOLOv4 /YOLOv5。 YOLOX使用 PyTorch开发，采用了Anchor-free机制、解耦头、Multi Positives、先进的标签分配策略和强数据增广等前沿技术...

输入的数据类型为torch.cuda.FloatTensor，说明输入数据在GPU中模型参数的数据类型为torch.cuda.FloatTensor，说明模型还在CPU里问题原因搞清楚了，模型没加载到CPU，在代码中加一行语句就可以了 model = model....

错误内容大概就是指输入类型是CPU（torch.FloatTensor），而参数类型是GPU（torch.cuda.FloatTensor）。关于数据类型的链接：官方链接首先，请先检查是否正确使用了CUDA。通常我们这样指定使用CUDA： device = ...

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

DeepStream社区

64

社区成员

230

社区内容

发帖

与我相关

我的任务

自动驾驶pytorch边缘计算个人社区陕西省·西安市

社区管理员

加入社区

近7日
近30日
至今

加载中

查看更多榜单

社区公告

暂无公告

试试用AI创作助手写篇文章吧

+ 用AI写文章