RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0!

[94268]

I get the following error using the inference.py code: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0!
When I debug, I get the following result:input = {devic:'cuda:0'},weight = device : cpu
I look forward to answering my questions at your convenience. Thank you very much!