报错：
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
分析原因：
cpu和cdua 使用的不一致，比如说你的模型和输入，一个在cpu上，一个在cuda上，导致这样的错误。
解决方法：
net = Model(cfg).to(device)
net.load_state_dict(torch.load(w2))
中的device，看输入的img是在cpu上还是cuda上，调整一下，要么是cpu要么是cuda
当模型和输入数据设备不一致，如CUDA与CPU混合使用时，会导致RuntimeError。解决办法是确保模型和输入都在同一设备上，例如通过`.to(device)`将它们全部转移到CPU或CUDA。检查`net=Model(cfg).to(device)`和加载权重的部分，确保设备变量`device`与输入图像设备相匹配。
扫一扫
2440
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
