I just found this Stack Overflow post. I understand now.

  

      stackoverflow.com
  

  
      
    
  


  How to get the device type of a pytorch module conveniently? 33



  python, gpu, pytorch



  asked by
  
  
    Kani
  
  on 03:09AM - 19 Nov 19 UTC