Skip to content
Snippets Groups Projects
Commit deb7dd15 authored by Neta Zmora's avatar Neta Zmora
Browse files

pytorch 0.4: begin the effort of centralizing the knowledge about which device to use

Eventually the application will pass a torch.device to the Scheduler.
Now we just create a default device in the constructor, and then use it.
parent 384f4740
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment