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

...全文

64 回复打赏收藏举报

写回复

回复

切换为时间正序

请发表友善的回复…

发表回复

相关推荐

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

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or module = torch.load("vincent_0.pth", map_location=torch.device("cpu"))

在使用torchsummary.summary时遇到Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same问题. 错误内容大概就是指输入类型是CPU（torch.FloatTensor），而参数类型是GPU（torch.cuda.FloatTensor）。两种用法： model = torchvision.models.alexnet() if vis_model:

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same 原因 Input type and weight type not be same fixed device = torch.device("cpu") model = MyModel() model.to(device) input.to(device) model(input) # cpu o

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor RuntimeError:输入类型(torch.FloatTensor)和权重类型(torch.cuda.FloatTensor)应该相同，或者输入应该是一个MKLDNN张

发帖

DeepStream社区

60

社区成员

228

社区内容

基于NVIDIA的DeepStream SDK开发技术交流社区。通过DeepStream可从USB/CSI/RTSP相机中获取流数据，用于快速构建具有AI能力的应用程序。

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

社区管理员

加入社区

帖子事件

创建了帖子

2022-10-26 09:42

社区公告

暂无公告