Answers for "model = to_device(Network(), device) next(model.parameters()).device #check the model weather it is in gpu or cpu"

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 "model = to_device(Network(), device) next(model.parameters()).device #check the model weather it is in gpu or cpu"