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

[Cheren15]

Hi, I am trying reproduce your work, but came across error as title showing. I have checked the location of '''model''' and '''imgs''' variables, even it shows to be cuda:0, however, the error still exists.
Do you have any clue about this, or I wondered if you can provide weights of trained 4 channel model so we can test with it?