Answers for "pytorch check if layer is on gpu"

how to check weather my model is on gpu in pytorch

next(network.parameters()).device

Posted by: Guest on June-28-2020

Code answers related to "pytorch check if layer is on gpu"