Closed Solitude1024 closed 3 months ago
作者你好,请问这个报错是cuda哪里用错了吗?
你好,从报错来看是你模型加载的权重是放在gpu上的,而输入的图像数据是放在cpu上的。将 retored = model(restored)(用于cpu测试)改为restored=model(restored.cuda())(用于gpu测试)即可。我刚刚更新了demo.py文件,感谢你的反馈