Why `criterion.cuda()` is not needed but `model.cuda()` is?

autograd
2

The impact of moving a module to cuda is actually to move all it’s parameters to cuda.
Criterion don’t have parameters in general, so it is not necessary to do it.

5 Likes