search

DeniJsonC / WaveNet

[PG 2023] WaveNet: Wave-Aware Image Enhancement
Other
78 stars 14 forks source link

运行问题 #11

Closed Solitude1024 closed 3 months ago

Solitude1024 commented 3 months ago

image image 作者你好，请问这个报错是cuda哪里用错了吗？

DeniJsonC commented 3 months ago

image image 作者你好，请问这个报错是cuda哪里用错了吗？

你好，从报错来看是你模型加载的权重是放在gpu上的，而输入的图像数据是放在cpu上的。将 retored = model(restored)(用于cpu测试)改为restored=model(restored.cuda())（用于gpu测试）即可。我刚刚更新了demo.py文件，感谢你的反馈