[Solved] RuntimeError: Input and parameter tensors are not at the same device, found input tensor at cpu and parameter tensor at cuda:0
Last Updated on 2021-06-28 by Clay
Today, when I load a model I trained with PyTorch and I want to use it to process some new data, I got a error report:
Read More »[Solved] RuntimeError: Input and parameter tensors are not at the same device, found input tensor at cpu and parameter tensor at cuda:0