pytorch
Making Sense of PyTorch's to(device) and map_location.
In a project I am working on, I had to push the model trained on a GPU into production. The problem started when I tried to make an inference script for the model and execute it in a CPU-only device to test the script. Habituated to the .to(device) method,
