(when checking argument for argument mat1 in method wrapper_addmm)

[ptrblck]

@Jun_Park

pl.Trainer moves the tensors (model) to the GPU during training and back to the CPU after training, if their locations are not explicitly specified.

That’s interesting. Do you know why that’s the case as it seems to be a bit unexpected (at least to me)?