Assume you have an nn module which was trained and saved to a .pkl file.
For inference, you need to load this model and decide whether you would like to run it on GPU or CPU. This is done by setting the argument “map_location” of “torch.load” to “cpu” for running on the CPU.
I think if you could give me any info about this error message then I would understand the issue better.