pytorch
is it okay to calculate loss in cpu and backpropagate in torch?
When using torch, while it is common to run the network in gpu, I wasn’t always so sure if it was mandatory for me to calculate the loss on the same gpu as the output at all times. If I Read more…