I tried to push the second img_out to GPU again, but still gives the same error information; But it can run pretty well on my computer.
For create the mean and std as CUDA tensors, could you please give me some hint? I am pretty new to pytorch and have no idea how to do it. Thanks!