Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

how to check weather my model is on gpu in pytorch

next(network.parameters()).device
Source by discuss.pytorch.org #
 
PREVIOUS NEXT
Tagged: #check #weather #model #gpu #pytorch
ADD COMMENT
Topic
Name
3+6 =