Answers for "setting device in torch"

0

setting device in torch

. You can use the tensor.to(device) command to move a tensor to a device. The .to() command is also used to move a whole model to a device, like in the post you linked to. to set cuda as your device if possible
Posted by: Guest on April-18-2021

Browse Popular Code Answers by Language