Skip to content

Comments

Added support for CPU devices#3

Open
Bebra777228 wants to merge 1 commit intojakeoneijk:mainfrom
Bebra777228:CPU_support
Open

Added support for CPU devices#3
Bebra777228 wants to merge 1 commit intojakeoneijk:mainfrom
Bebra777228:CPU_support

Conversation

@Bebra777228
Copy link

The following error appeared during inference on the CPU device:

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available () is False. If you are running on a CPU-only machine, please use torch.load with map_location = torch.device ('cpu') to map your storages to the CPU.

This PR fixes it.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant