albanD:

So you already compute the loss on a single device no?


Yes, the loss is computed in a single device. But I really don’t understand why it still raise that error.
I’m very newbie in PyTorch. Could you please guide me what I’ve done wrong. Thank you.